Searched defs:scale_data (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | fused_batch_norm_ex_op_test.cc | 137 const Tensor& input_data, const Tensor& scale_data, in RunFusedBatchNorm()
225 const Tensor& input_data, const Tensor& scale_data, in RunFusedBatchNormEx()
445 const Tensor& scale_data, const Tensor& offset_data, in VerifyFusedBatchNormEx()
448 FusedBatchNormGradOutputs* bwd) { in VerifyFusedBatchNormEx()
|
| D | conv_ops_test.cc | 611 const Tensor& scale_data, const Tensor& offset_data, in RunConv2DWithBatchNorm()
640 const Tensor& scale_data, const Tensor& offset_data, in RunConv2DWithBatchNormAndActivation()
897 const Tensor& scale_data, const Tensor& offset_data,
920 const Tensor& scale_data, const Tensor& offset_data,
930 const Tensor& scale_data, const Tensor& offset_data,
931 const Tensor& mean_data, const Tensor& variance_data, Tensor* out) { in __anon87b5fd160902()
|
| D | fused_eigen_output_kernels.h | 148 const T* scale_data = nullptr; member
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | fused_batchnorm_reserve_space_test.cc | 94 Tensor scale_data(DT_FLOAT, TensorShape({10})); in TEST() local
|
| /external/tensorflow/tensorflow/core/kernels/image/ |
| D | adjust_saturation_op.cc | 238 const float* scale_data = scale->flat<float>().data(); in DoCompute() local
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | batch_normalization_test.cc | 630 std::unique_ptr<GlobalData> scale_data = in XLA_TEST_P() local
731 std::unique_ptr<GlobalData> scale_data = in XLA_TEST_P() local
916 std::unique_ptr<GlobalData> scale_data = in XLA_TEST_P() local
|
| /external/tensorflow/tensorflow/stream_executor/ |
| D | dnn.h | 2439 const DeviceMemory<float>& scale_data, in DoFusedBatchNormActivationInference()
2452 const DeviceMemory<float>& scale_data, in DoFusedBatchNormActivationInference()
2493 const DeviceMemory<float>& scale_data, in DoFusedBatchNormActivationForward()
2506 const DeviceMemory<float>& scale_data, in DoFusedBatchNormActivationForward()
2553 const DeviceMemory<float>& scale_data, in DoFusedBatchNormActivationBackward()
2571 const DeviceMemory<float>& scale_data, in DoFusedBatchNormActivationBackward()
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_dnn.cc | 4620 const DeviceMemory<U>& scale_data, const DeviceMemory<U>& offset_data, in DoFusedBatchNormActivationInferenceImpl()
4695 const DeviceMemory<float>& scale_data, in DoFusedBatchNormActivationInference()
4711 const DeviceMemory<float>& scale_data, in DoFusedBatchNormActivationInference()
4729 const DeviceMemory<U>& scale_data, const DeviceMemory<U>& offset_data, in DoFusedBatchNormActivationForwardImpl()
4807 const DeviceMemory<float>& scale_data, in DoFusedBatchNormActivationForward()
4824 const DeviceMemory<float>& scale_data, in DoFusedBatchNormActivationForward()
4846 const DeviceMemory<U>& scale_data, const DeviceMemory<U>& offset_data, in DoFusedBatchNormActivationBackwardImpl()
4927 const DeviceMemory<float>& scale_data, in DoFusedBatchNormActivationBackward()
4950 const DeviceMemory<float>& scale_data, in DoFusedBatchNormActivationBackward()
|