| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | fused_batch_norm_ex_op_test.cc | 137 const Tensor& input_data, const Tensor& scale_data, in RunFusedBatchNorm()
224 const Tensor& input_data, const Tensor& scale_data, in RunFusedBatchNormEx()
466 const Tensor& scale_data, const Tensor& offset_data, in VerifyFusedBatchNormEx()
469 FusedBatchNormGradOutputs* bwd) { in VerifyFusedBatchNormEx()
|
| H A 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 __anona74cd8200902()
|
| H A D | fused_eigen_output_kernels.h | 149 const T* scale_data = nullptr; member
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/ |
| H A D | dnn.h | 2465 const DeviceMemory<float>& scale_data, in DoFusedBatchNormActivationInference()
2478 const DeviceMemory<float>& scale_data, in DoFusedBatchNormActivationInference()
2519 const DeviceMemory<float>& scale_data, in DoFusedBatchNormActivationForward()
2532 const DeviceMemory<float>& scale_data, in DoFusedBatchNormActivationForward()
2579 const DeviceMemory<float>& scale_data, in DoFusedBatchNormActivationBackward()
2597 const DeviceMemory<float>& scale_data, in DoFusedBatchNormActivationBackward()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/tests/ |
| H A D | batch_normalization_test.cc | 595 std::unique_ptr<GlobalData> scale_data = in XLA_TEST_P() local
696 std::unique_ptr<GlobalData> scale_data = in XLA_TEST_P() local
881 std::unique_ptr<GlobalData> scale_data = in XLA_TEST_P() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/stream_executor/rocm/ |
| H A D | rocm_dnn.cc | 4669 const DeviceMemory<U>& scale_data, const DeviceMemory<U>& offset_data, in DoFusedBatchNormActivationInferenceImpl()
4744 const DeviceMemory<float>& scale_data, in DoFusedBatchNormActivationInference()
4760 const DeviceMemory<float>& scale_data, in DoFusedBatchNormActivationInference()
4778 const DeviceMemory<U>& scale_data, const DeviceMemory<U>& offset_data, in DoFusedBatchNormActivationForwardImpl()
4856 const DeviceMemory<float>& scale_data, in DoFusedBatchNormActivationForward()
4873 const DeviceMemory<float>& scale_data, in DoFusedBatchNormActivationForward()
4895 const DeviceMemory<U>& scale_data, const DeviceMemory<U>& offset_data, in DoFusedBatchNormActivationBackwardImpl()
4976 const DeviceMemory<float>& scale_data, in DoFusedBatchNormActivationBackward()
4999 const DeviceMemory<float>& scale_data, in DoFusedBatchNormActivationBackward()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/ |
| H A D | fused_batchnorm_reserve_space_test.cc | 94 Tensor scale_data(DT_FLOAT, TensorShape({10})); in TEST() local
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | EmbeddingBag.cpp | 525 auto* scale_data = scale.const_data_ptr<data_t>(); in index_select_scale_add() local
567 auto* scale_data = scale.const_data_ptr<data_t>(); in index_select_scale_add() local
756 auto* scale_data = scale.const_data_ptr<float>(); in index_select_scale_add() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/image/ |
| H A D | adjust_saturation_op.cc | 238 const float* scale_data = scale->flat<float>().data(); in DoCompute() local
|
| /aosp_15_r20/external/executorch/kernels/quantized/cpu/ |
| H A D | op_quantize.cpp | 293 const double* scale_data = scale.const_data_ptr<double>(); in quantize_per_channel_out() local
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/passes/onnx/ |
| H A D | unpack_quantized_weights.cpp | 112 std::vector<float> scale_data; in CreateQuantizedWeights() local
|
| /aosp_15_r20/external/executorch/backends/xnnpack/runtime/ |
| H A D | XNNCompiler.cpp | 402 const uint16_t* scale_data = nullptr; in defineTensor() local
|