| /external/tensorflow/tensorflow/contrib/lite/kernels/ |
| H A D | bidirectional_sequence_rnn_test.cc | 631 constexpr std::initializer_list<float> biases = { member in namespace:tflite::__anon25925
772 rnn.SetFwBias(biases);
773 rnn.SetBwBias(biases);
811 rnn.SetFwBias(biases);
812 rnn.SetBwBias(biases);
878 rnn.SetFwBias(biases);
879 rnn.SetBwBias(biases);
|
| /external/tensorflow/tensorflow/examples/tutorials/word2vec/ |
| H A D | word2vec_basic.py | 201 with tf.name_scope('biases'):
213 biases=nce_biases, variable
|
| /external/tensorflow/tensorflow/contrib/cudnn_rnn/kernels/ |
| H A D | cudnn_rnn_ops.cc | 769 OpInputList biases; variable
770 OP_REQUIRES_OK(context, context->input_list("biases", &biases));
771 RestoreParams<T>(biases, rnn_desc->ParamsBiasRegions(), &output_ptr,
|
| /external/tensorflow/tensorflow/stream_executor/ |
| H A D | stream.cc | 423 const DeviceMemory<float> &biases, dnn::ActivationMode activation_mode,
430 PARAM(bias_descriptor), PARAM(biases), PARAM(activation_mode),
438 side_input_data, side_input_scale, bias_descriptor, biases,
456 const DeviceMemory<Eigen::half> &biases,
464 PARAM(bias_descriptor), PARAM(biases), PARAM(activation_mode),
472 side_input_data, side_input_scale, bias_descriptor, biases,
490 const DeviceMemory<float> &biases, dnn::ActivationMode activation_mode,
497 PARAM(bias_descriptor), PARAM(biases), PARAM(activation_mode),
505 side_input_data, side_input_scale, bias_descriptor, biases,
577 const DeviceMemory<float> &biases, dn
415 ThenFusedConvolveWithScratch( const dnn::BatchDescriptor &conv_input_descriptor, const DeviceMemory<int8> &conv_input_data, float conv_input_scale, const dnn::FilterDescriptor &filter_descriptor, const DeviceMemory<int8> &filter_data, const dnn::ConvolutionDescriptor &convolution_descriptor, const DeviceMemory<int8> &side_input_data, float side_input_scale, const dnn::BatchDescriptor &bias_descriptor, const DeviceMemory<float> &biases, dnn::ActivationMode activation_mode, const dnn::BatchDescriptor &output_descriptor, DeviceMemory<int8> *output, ScratchAllocator *scratch_allocator) argument
448 ThenFusedConvolveWithScratch( const dnn::BatchDescriptor &conv_input_descriptor, const DeviceMemory<Eigen::half> &conv_input_data, float conv_input_scale, const dnn::FilterDescriptor &filter_descriptor, const DeviceMemory<Eigen::half> &filter_data, const dnn::ConvolutionDescriptor &convolution_descriptor, const DeviceMemory<Eigen::half> &side_input_data, float side_input_scale, const dnn::BatchDescriptor &bias_descriptor, const DeviceMemory<Eigen::half> &biases, dnn::ActivationMode activation_mode, const dnn::BatchDescriptor &output_descriptor, DeviceMemory<Eigen::half> *output, ScratchAllocator *scratch_allocator) argument
482 ThenFusedConvolveWithScratch( const dnn::BatchDescriptor &conv_input_descriptor, const DeviceMemory<float> &conv_input_data, float conv_input_scale, const dnn::FilterDescriptor &filter_descriptor, const DeviceMemory<float> &filter_data, const dnn::ConvolutionDescriptor &convolution_descriptor, const DeviceMemory<float> &side_input_data, float side_input_scale, const dnn::BatchDescriptor &bias_descriptor, const DeviceMemory<float> &biases, dnn::ActivationMode activation_mode, const dnn::BatchDescriptor &output_descriptor, DeviceMemory<float> *output, ScratchAllocator *scratch_allocator) argument
569 ThenFusedConvolveWithAlgorithm( const dnn::BatchDescriptor &conv_input_descriptor, const DeviceMemory<float> &conv_input_data, float conv_input_scale, const dnn::FilterDescriptor &filter_descriptor, const DeviceMemory<float> &filter_data, const dnn::ConvolutionDescriptor &convolution_descriptor, const DeviceMemory<float> &side_input_data, float side_input_scale, const dnn::BatchDescriptor &bias_descriptor, const DeviceMemory<float> &biases, dnn::ActivationMode activation_mode, const dnn::BatchDescriptor &output_descriptor, DeviceMemory<float> *output, ScratchAllocator *scratch_allocator, const dnn::AlgorithmConfig &algorithm_config, dnn::ProfileResult *output_profile_result) argument
607 ThenFusedConvolveWithAlgorithm( const dnn::BatchDescriptor &conv_input_descriptor, const DeviceMemory<Eigen::half> &conv_input_data, float conv_input_scale, const dnn::FilterDescriptor &filter_descriptor, const DeviceMemory<Eigen::half> &filter_data, const dnn::ConvolutionDescriptor &convolution_descriptor, const DeviceMemory<Eigen::half> &side_input_data, float side_input_scale, const dnn::BatchDescriptor &bias_descriptor, const DeviceMemory<Eigen::half> &biases, dnn::ActivationMode activation_mode, const dnn::BatchDescriptor &output_descriptor, DeviceMemory<Eigen::half> *output, ScratchAllocator *scratch_allocator, const dnn::AlgorithmConfig &algorithm_config, dnn::ProfileResult *output_profile_result) argument
646 ThenFusedConvolveWithAlgorithm( const dnn::BatchDescriptor &conv_input_descriptor, const DeviceMemory<int8> &conv_input_data, float conv_input_scale, const dnn::FilterDescriptor &filter_descriptor, const DeviceMemory<int8> &filter_data, const dnn::ConvolutionDescriptor &convolution_descriptor, const DeviceMemory<int8> &side_input_data, float side_input_scale, const dnn::BatchDescriptor &bias_descriptor, const DeviceMemory<float> &biases, dnn::ActivationMode activation_mode, const dnn::BatchDescriptor &output_descriptor, DeviceMemory<int8> *output, ScratchAllocator *scratch_allocator, const dnn::AlgorithmConfig &algorithm_config, dnn::ProfileResult *output_profile_result) argument
746 ThenFusedConvolve( const dnn::BatchDescriptor &conv_input_descriptor, const DeviceMemory<int8> &conv_input_data, float conv_input_scale, const dnn::FilterDescriptor &filter_descriptor, const DeviceMemory<int8> &filter_data, const dnn::ConvolutionDescriptor &convolution_descriptor, const DeviceMemory<int8> &side_input_data, float side_input_scale, const dnn::BatchDescriptor &bias_descriptor, const DeviceMemory<float> &biases, dnn::ActivationMode activation_mode, const dnn::BatchDescriptor &output_descriptor, DeviceMemory<int8> *output) argument
1248 ThenBiasAdd(const DeviceMemory<float> &input_data, const DeviceMemory<float> &biases, const dnn::BatchDescriptor &dimensions, DeviceMemory<float> *output_data) argument
[all...] |
| H A D | dnn.h | 323 // Return the number of biases required to fully connect to an output layer
431 // Returns the number of biases required as parameters for a convolution
995 // We provide several variants with different types for inputs, biases and
1010 // biases: un-owned device memory region containing biases to add to the
1054 const DeviceMemory<double>& biases, dnn::ActivationMode activation_mode,
1071 const DeviceMemory<float>& biases, dnn::ActivationMode activation_mode,
1089 const DeviceMemory<Eigen::half>& biases,
1109 const DeviceMemory<float>& biases, dnn::ActivationMode activation_mode,
1450 // Adds biases t
1046 DoFusedConvolve( Stream* stream, const dnn::BatchDescriptor& conv_input_descriptor, const DeviceMemory<double>& conv_input_data, double conv_input_scale, const dnn::FilterDescriptor& filter_descriptor, const DeviceMemory<double>& filter_data, const dnn::ConvolutionDescriptor& convolution_descriptor, const DeviceMemory<double>& side_input_data, double side_input_scale, const dnn::BatchDescriptor& bias_descriptor, const DeviceMemory<double>& biases, dnn::ActivationMode activation_mode, const dnn::BatchDescriptor& output_descriptor, DeviceMemory<double>* output_data, ScratchAllocator* scratch_allocator, const dnn::AlgorithmConfig& algorithm_config, dnn::ProfileResult* output_profile_result) argument
1063 DoFusedConvolve( Stream* stream, const dnn::BatchDescriptor& conv_input_descriptor, const DeviceMemory<float>& conv_input_data, float conv_input_scale, const dnn::FilterDescriptor& filter_descriptor, const DeviceMemory<float>& filter_data, const dnn::ConvolutionDescriptor& convolution_descriptor, const DeviceMemory<float>& side_input_data, float side_input_scale, const dnn::BatchDescriptor& bias_descriptor, const DeviceMemory<float>& biases, dnn::ActivationMode activation_mode, const dnn::BatchDescriptor& output_descriptor, DeviceMemory<float>* output_data, ScratchAllocator* scratch_allocator, const dnn::AlgorithmConfig& algorithm_config, dnn::ProfileResult* output_profile_result) argument
1081 DoFusedConvolve( Stream* stream, const dnn::BatchDescriptor& conv_input_descriptor, const DeviceMemory<Eigen::half>& conv_input_data, float conv_input_scale, const dnn::FilterDescriptor& filter_descriptor, const DeviceMemory<Eigen::half>& filter_data, const dnn::ConvolutionDescriptor& convolution_descriptor, const DeviceMemory<Eigen::half>& side_input_data, float side_input_scale, const dnn::BatchDescriptor& bias_descriptor, const DeviceMemory<Eigen::half>& biases, dnn::ActivationMode activation_mode, const dnn::BatchDescriptor& output_descriptor, DeviceMemory<Eigen::half>* output_data, ScratchAllocator* scratch_allocator, const dnn::AlgorithmConfig& algorithm_config, dnn::ProfileResult* output_profile_result) argument
1101 DoFusedConvolve( Stream* stream, const dnn::BatchDescriptor& conv_input_descriptor, const DeviceMemory<int8>& conv_input_data, float conv_input_scale, const dnn::FilterDescriptor& filter_descriptor, const DeviceMemory<int8>& filter_data, const dnn::ConvolutionDescriptor& convolution_descriptor, const DeviceMemory<int8>& side_input_data, float side_input_scale, const dnn::BatchDescriptor& bias_descriptor, const DeviceMemory<float>& biases, dnn::ActivationMode activation_mode, const dnn::BatchDescriptor& output_descriptor, DeviceMemory<int8>* output_data, ScratchAllocator* scratch_allocator, const dnn::AlgorithmConfig& algorithm_config, dnn::ProfileResult* output_profile_result) argument
[all...] |
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| H A D | cuda_dnn.cc | 1223 // Create the weights and biases into the params buffer
2510 const DeviceMemory<BiasType>& biases, dnn::ActivationMode activation_mode,
2596 << "\nbiases.opaque() = " << biases.opaque()
2608 /*biasDesc=*/bias_nd.handle(), /*bias=*/biases.opaque(),
2976 const DeviceMemory<double>& biases, dnn::ActivationMode activation_mode,
2985 side_input_scale, bias_descriptor, biases, activation_mode,
2998 const DeviceMemory<float>& biases, dnn::ActivationMode activation_mode,
3007 side_input_scale, bias_descriptor, biases, activation_mode,
3020 const DeviceMemory<Eigen::half>& biases,
3030 side_input_scale, bias_descriptor, biases, activation_mod
2502 DoFusedConvolveImpl( Stream* stream, const dnn::BatchDescriptor& conv_input_descriptor, const DeviceMemory<Type>& conv_input_data, ScaleType conv_input_scale, const dnn::FilterDescriptor& filter_descriptor, const DeviceMemory<Type>& filter_data, const dnn::ConvolutionDescriptor& convolution_descriptor, const DeviceMemory<Type>& side_input_data, ScaleType side_input_scale, const dnn::BatchDescriptor& bias_descriptor, const DeviceMemory<BiasType>& biases, dnn::ActivationMode activation_mode, const dnn::BatchDescriptor& output_descriptor, DeviceMemory<Type>* output_data, ScratchAllocator* scratch_allocator, const dnn::AlgorithmConfig& algorithm_config, dnn::ProfileResult* output_profile_result) argument
2968 DoFusedConvolve( Stream* stream, const dnn::BatchDescriptor& conv_input_descriptor, const DeviceMemory<double>& conv_input_data, double conv_input_scale, const dnn::FilterDescriptor& filter_descriptor, const DeviceMemory<double>& filter_data, const dnn::ConvolutionDescriptor& convolution_descriptor, const DeviceMemory<double>& side_input_data, double side_input_scale, const dnn::BatchDescriptor& bias_descriptor, const DeviceMemory<double>& biases, dnn::ActivationMode activation_mode, const dnn::BatchDescriptor& output_descriptor, DeviceMemory<double>* output_data, ScratchAllocator* scratch_allocator, const dnn::AlgorithmConfig& algorithm_config, dnn::ProfileResult* output_profile_result) argument
2990 DoFusedConvolve( Stream* stream, const dnn::BatchDescriptor& conv_input_descriptor, const DeviceMemory<float>& conv_input_data, float conv_input_scale, const dnn::FilterDescriptor& filter_descriptor, const DeviceMemory<float>& filter_data, const dnn::ConvolutionDescriptor& convolution_descriptor, const DeviceMemory<float>& side_input_data, float side_input_scale, const dnn::BatchDescriptor& bias_descriptor, const DeviceMemory<float>& biases, dnn::ActivationMode activation_mode, const dnn::BatchDescriptor& output_descriptor, DeviceMemory<float>* output_data, ScratchAllocator* scratch_allocator, const dnn::AlgorithmConfig& algorithm_config, dnn::ProfileResult* output_profile_result) argument
3012 DoFusedConvolve( Stream* stream, const dnn::BatchDescriptor& conv_input_descriptor, const DeviceMemory<Eigen::half>& conv_input_data, float conv_input_scale, const dnn::FilterDescriptor& filter_descriptor, const DeviceMemory<Eigen::half>& filter_data, const dnn::ConvolutionDescriptor& convolution_descriptor, const DeviceMemory<Eigen::half>& side_input_data, float side_input_scale, const dnn::BatchDescriptor& bias_descriptor, const DeviceMemory<Eigen::half>& biases, dnn::ActivationMode activation_mode, const dnn::BatchDescriptor& output_descriptor, DeviceMemory<Eigen::half>* output_data, ScratchAllocator* scratch_allocator, const dnn::AlgorithmConfig& algorithm_config, dnn::ProfileResult* output_profile_result) argument
3035 DoFusedConvolve( Stream* stream, const dnn::BatchDescriptor& conv_input_descriptor, const DeviceMemory<int8>& conv_input_data, float conv_input_scale, const dnn::FilterDescriptor& filter_descriptor, const DeviceMemory<int8>& filter_data, const dnn::ConvolutionDescriptor& convolution_descriptor, const DeviceMemory<int8>& side_input_data, float side_input_scale, const dnn::BatchDescriptor& bias_descriptor, const DeviceMemory<float>& biases, dnn::ActivationMode activation_mode, const dnn::BatchDescriptor& output_descriptor, DeviceMemory<int8>* output_data, ScratchAllocator* scratch_allocator, const dnn::AlgorithmConfig& algorithm_config, dnn::ProfileResult* output_profile_result) argument
3827 DoBiasAdd(Stream* stream, const DeviceMemory<float>& input_data, const DeviceMemory<float>& biases, const dnn::BatchDescriptor& dimensions, DeviceMemory<float>* output_data) argument
[all...] |