Searched defs:scale_data (Results 1 – 5 of 5) sorted by relevance
/external/tensorflow/tensorflow/compiler/tf2xla/ |
D | fused_batchnorm_reserve_space_test.cc | 80 Tensor scale_data(DT_FLOAT, TensorShape({10})); in TEST() local
|
/external/tensorflow/tensorflow/core/kernels/ |
D | adjust_saturation_op.cc | 236 const float* scale_data = scale->flat<float>().data(); in DoCompute() local
|
D | conv_ops_test.cc | 650 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/ |
D | batch_normalization_test.cc | 532 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/ |
D | rocm_dnn.cc | 4134 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()
|