| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | fused_batchnorm_reserve_space_test.cc | 80 Tensor scale_data(DT_FLOAT, TensorShape({10})); in TEST() local
81 FillZeros(&scale_data); in TEST()
83 Const(root.WithOpName("scale"), Input::Initializer(scale_data)); in TEST()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | conv_ops_test.cc | 544 const Tensor& scale_data, const Tensor& offset_data,
650 const Tensor& scale_data, in RunConv2DWithBatchNorm() argument
668 ops::Const(root.WithOpName("scale"), Input::Initializer(scale_data)), in RunConv2DWithBatchNorm()
679 const Tensor& scale_data, const Tensor& offset_data, in RunConv2DWithBatchNormAndRelu() argument
695 ops::Const(root.WithOpName("scale"), Input::Initializer(scale_data)), in RunConv2DWithBatchNormAndRelu()
893 const Tensor& scale_data, const Tensor& offset_data, in VerifyConv2DWithBatchNorm()
896 RunConv2DWithBatchNorm(input_data, filter_data, scale_data, in VerifyConv2DWithBatchNorm()
902 const Tensor& scale_data, const Tensor& offset_data, in VerifyConv2DWithBatchNorm() argument
906 {scale_data, offset_data, mean_data, variance_data}, in VerifyConv2DWithBatchNorm()
923 const Tensor& scale_data, const Tensor& offset_data, in VerifyConv2DWithBatchNormAndRelu() argument
[all …]
|
| D | adjust_saturation_op.cc | 236 const float* scale_data = scale->flat<float>().data(); in DoCompute() local
239 scale_data, output_data); in DoCompute()
|
| D | conv_ops_fused_impl.h | 375 const T* scale_data = nullptr;
426 InitDataPtr(scale, &args->scale_data);
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_dnn.h | 549 const DeviceMemory<float>& scale_data,
560 const DeviceMemory<float>& scale_data,
571 const DeviceMemory<float>& scale_data,
582 const DeviceMemory<float>& scale_data,
595 const DeviceMemory<float>& scale_data,
611 const DeviceMemory<float>& scale_data,
724 const DeviceMemory<U>& scale_data, const DeviceMemory<U>& offset_data,
735 const DeviceMemory<U>& scale_data, const DeviceMemory<U>& offset_data,
751 const DeviceMemory<U>& scale_data, const DeviceMemory<U>& offset_data,
|
| D | rocm_dnn.cc | 4134 const DeviceMemory<U>& scale_data, const DeviceMemory<U>& offset_data, in DoFusedBatchNormActivationInferenceImpl() argument
4169 scale_data.opaque(), offset_data.opaque(), mean_data.opaque(), in DoFusedBatchNormActivationInferenceImpl()
4209 const DeviceMemory<float>& scale_data, in DoFusedBatchNormActivationInference() argument
4217 scale_offset_mean_variance_descriptor, scale_data, offset_data, mean_data, in DoFusedBatchNormActivationInference()
4225 const DeviceMemory<float>& scale_data, in DoFusedBatchNormActivationInference() argument
4233 scale_offset_mean_variance_descriptor, scale_data, offset_data, mean_data, in DoFusedBatchNormActivationInference()
4243 const DeviceMemory<U>& scale_data, const DeviceMemory<U>& offset_data, in DoFusedBatchNormActivationForwardImpl() argument
4280 scale_data.opaque(), offset_data.opaque(), batch_mean_data->opaque(), in DoFusedBatchNormActivationForwardImpl()
4321 const DeviceMemory<float>& scale_data, in DoFusedBatchNormActivationForward() argument
4329 scale_offset_mean_variance_descriptor, scale_data, offset_data, epsilon, in DoFusedBatchNormActivationForward()
[all …]
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | batch_normalization_test.cc | 532 std::unique_ptr<GlobalData> scale_data = in XLA_TEST_P() local
546 {input_data.get(), scale_data.get(), offset_data.get()}, in XLA_TEST_P()
633 std::unique_ptr<GlobalData> scale_data = in XLA_TEST_P() local
653 {input_data.get(), scale_data.get(), offset_data.get(), mean_data.get(), in XLA_TEST_P()
818 std::unique_ptr<GlobalData> scale_data = in XLA_TEST_P() local
840 {input_data.get(), scale_data.get(), mean_data.get(), in XLA_TEST_P()
|
| /external/tensorflow/tensorflow/stream_executor/ |
| D | dnn.h | 2396 const DeviceMemory<float>& scale_data,
2409 const DeviceMemory<float>& scale_data,
2450 const DeviceMemory<float>& scale_data,
2463 const DeviceMemory<float>& scale_data,
2510 const DeviceMemory<float>& scale_data,
2528 const DeviceMemory<float>& scale_data,
|