| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | fused_batchnorm_reserve_space_test.cc | 94 Tensor scale_data(DT_FLOAT, TensorShape({10})); in TEST() local
95 FillZeros(&scale_data); in TEST()
97 Const(root.WithOpName("scale"), Input::Initializer(scale_data)); in TEST()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | fused_batch_norm_ex_op_test.cc | 70 const Tensor& scale_data, const Tensor& offset_data,
137 const Tensor& input_data, const Tensor& scale_data, in RunFusedBatchNorm() argument
152 ops::Const(root.WithOpName("scale"), Input::Initializer(scale_data)); in RunFusedBatchNorm()
225 const Tensor& input_data, const Tensor& scale_data, in RunFusedBatchNormEx() argument
244 ops::Const(root.WithOpName("scale"), Input::Initializer(scale_data)); in RunFusedBatchNormEx()
433 const Tensor& scale_data, const Tensor& offset_data, in VerifyFusedBatchNormEx()
437 this->RunFusedBatchNorm(y_backprop, input_data, scale_data, in VerifyFusedBatchNormEx()
445 const Tensor& scale_data, const Tensor& offset_data, in VerifyFusedBatchNormEx() argument
449 this->RunFusedBatchNormEx(y_backprop, input_data, scale_data, in VerifyFusedBatchNormEx()
|
| D | conv_ops_test.cc | 547 const Tensor& scale_data, const Tensor& offset_data,
668 const Tensor& scale_data, const Tensor& offset_data, in RunConv2DWithBatchNorm() argument
686 ops::Const(root.WithOpName("scale"), Input::Initializer(scale_data)), in RunConv2DWithBatchNorm()
697 const Tensor& scale_data, const Tensor& offset_data, in RunConv2DWithBatchNormAndActivation() argument
716 ops::Const(root.WithOpName("scale"), Input::Initializer(scale_data)), in RunConv2DWithBatchNormAndActivation()
944 const Tensor& scale_data, const Tensor& offset_data,
946 RunConv2DWithBatchNorm(input_data, filter_data, scale_data, in __anone7ca09510602()
954 const Tensor& scale_data, const Tensor& offset_data, argument
957 {scale_data, offset_data, mean_data, variance_data}, in __anone7ca09510702()
977 const Tensor& scale_data, const Tensor& offset_data, argument
[all …]
|
| D | fused_eigen_output_kernels.h | 148 const T* scale_data = nullptr; member
418 args->scale_data = data_ptr(scale);
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_dnn.h | 601 const DeviceMemory<float>& scale_data,
612 const DeviceMemory<float>& scale_data,
623 const DeviceMemory<float>& scale_data,
634 const DeviceMemory<float>& scale_data,
647 const DeviceMemory<float>& scale_data,
663 const DeviceMemory<float>& scale_data,
799 const DeviceMemory<U>& scale_data, const DeviceMemory<U>& offset_data,
810 const DeviceMemory<U>& scale_data, const DeviceMemory<U>& offset_data,
826 const DeviceMemory<U>& scale_data, const DeviceMemory<U>& offset_data,
|
| D | rocm_dnn.cc | 4730 const DeviceMemory<U>& scale_data, const DeviceMemory<U>& offset_data, in DoFusedBatchNormActivationInferenceImpl() argument
4765 scale_data.opaque(), offset_data.opaque(), mean_data.opaque(), in DoFusedBatchNormActivationInferenceImpl()
4805 const DeviceMemory<float>& scale_data, in DoFusedBatchNormActivationInference() argument
4813 scale_offset_mean_variance_descriptor, scale_data, offset_data, mean_data, in DoFusedBatchNormActivationInference()
4821 const DeviceMemory<float>& scale_data, in DoFusedBatchNormActivationInference() argument
4829 scale_offset_mean_variance_descriptor, scale_data, offset_data, mean_data, in DoFusedBatchNormActivationInference()
4839 const DeviceMemory<U>& scale_data, const DeviceMemory<U>& offset_data, in DoFusedBatchNormActivationForwardImpl() argument
4876 scale_data.opaque(), offset_data.opaque(), batch_mean_data->opaque(), in DoFusedBatchNormActivationForwardImpl()
4917 const DeviceMemory<float>& scale_data, in DoFusedBatchNormActivationForward() argument
4925 scale_offset_mean_variance_descriptor, scale_data, offset_data, epsilon, in DoFusedBatchNormActivationForward()
[all …]
|
| /external/tensorflow/tensorflow/core/kernels/image/ |
| D | adjust_saturation_op.cc | 238 const float* scale_data = scale->flat<float>().data(); in DoCompute() local
241 scale_data, output_data); in DoCompute()
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | batch_normalization_test.cc | 630 std::unique_ptr<GlobalData> scale_data = in XLA_TEST_P() local
644 {input_data.get(), scale_data.get(), offset_data.get()}, in XLA_TEST_P()
731 std::unique_ptr<GlobalData> scale_data = in XLA_TEST_P() local
751 {input_data.get(), scale_data.get(), offset_data.get(), mean_data.get(), in XLA_TEST_P()
916 std::unique_ptr<GlobalData> scale_data = in XLA_TEST_P() local
938 {input_data.get(), scale_data.get(), mean_data.get(), in XLA_TEST_P()
|
| /external/tensorflow/tensorflow/stream_executor/ |
| D | dnn.h | 2531 const DeviceMemory<float>& scale_data, in DoFusedBatchNormActivationInference() argument
2544 const DeviceMemory<float>& scale_data, in DoFusedBatchNormActivationInference() argument
2585 const DeviceMemory<float>& scale_data, in DoFusedBatchNormActivationForward() argument
2598 const DeviceMemory<float>& scale_data, in DoFusedBatchNormActivationForward() argument
2645 const DeviceMemory<float>& scale_data, in DoFusedBatchNormActivationBackward() argument
2663 const DeviceMemory<float>& scale_data, in DoFusedBatchNormActivationBackward() argument
|