Home
last modified time | relevance | path

Searched defs:scale_data (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2xla/
Dfused_batchnorm_reserve_space_test.cc80 Tensor scale_data(DT_FLOAT, TensorShape({10})); in TEST() local
/external/tensorflow/tensorflow/core/kernels/
Dadjust_saturation_op.cc236 const float* scale_data = scale->flat<float>().data(); in DoCompute() local
Dconv_ops_test.cc650 const Tensor& scale_data, in RunConv2DWithBatchNorm()
679 const Tensor& scale_data, const Tensor& offset_data, in RunConv2DWithBatchNormAndRelu()
902 const Tensor& scale_data, const Tensor& offset_data, in VerifyConv2DWithBatchNorm()
923 const Tensor& scale_data, const Tensor& offset_data, in VerifyConv2DWithBatchNormAndRelu()
933 const Tensor& scale_data, const Tensor& offset_data, in VerifyConv2DWithBatchNormAndRelu()
935 Tensor* out) { in VerifyConv2DWithBatchNormAndRelu()
/external/tensorflow/tensorflow/compiler/xla/tests/
Dbatch_normalization_test.cc532 std::unique_ptr<GlobalData> scale_data = in XLA_TEST_P() local
633 std::unique_ptr<GlobalData> scale_data = in XLA_TEST_P() local
818 std::unique_ptr<GlobalData> scale_data = in XLA_TEST_P() local
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_dnn.cc4134 const DeviceMemory<U>& scale_data, const DeviceMemory<U>& offset_data, in DoFusedBatchNormActivationInferenceImpl()
4209 const DeviceMemory<float>& scale_data, in DoFusedBatchNormActivationInference()
4225 const DeviceMemory<float>& scale_data, in DoFusedBatchNormActivationInference()
4243 const DeviceMemory<U>& scale_data, const DeviceMemory<U>& offset_data, in DoFusedBatchNormActivationForwardImpl()
4321 const DeviceMemory<float>& scale_data, in DoFusedBatchNormActivationForward()
4338 const DeviceMemory<float>& scale_data, in DoFusedBatchNormActivationForward()
4360 const DeviceMemory<U>& scale_data, const DeviceMemory<U>& offset_data, in DoFusedBatchNormActivationBackwardImpl()
4441 const DeviceMemory<float>& scale_data, in DoFusedBatchNormActivationBackward()
4464 const DeviceMemory<float>& scale_data, in DoFusedBatchNormActivationBackward()