Home
last modified time | relevance | path

Searched defs:offset_data (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dfused_eigen_output_kernels.h129 const T* offset_data = nullptr; member
246 const T* offset_data; member
Dfused_batch_norm_ex_op_test.cc138 const Tensor& offset_data, const Tensor& mean_data, in RunFusedBatchNorm()
226 const Tensor& offset_data, const Tensor& mean_data, in RunFusedBatchNormEx()
445 const Tensor& scale_data, const Tensor& offset_data, in VerifyFusedBatchNormEx()
448 FusedBatchNormGradOutputs* bwd) { in VerifyFusedBatchNormEx()
Dconv_ops_test.cc664 const Tensor& scale_data, const Tensor& offset_data, in RunConv2DWithBatchNorm()
693 const Tensor& scale_data, const Tensor& offset_data, in RunConv2DWithBatchNormAndActivation()
947 const Tensor& scale_data, const Tensor& offset_data,
970 const Tensor& scale_data, const Tensor& offset_data,
980 const Tensor& scale_data, const Tensor& offset_data,
981 const Tensor& mean_data, const Tensor& variance_data, Tensor* out) { in __anon10aa6af50902()
/external/tensorflow/tensorflow/compiler/tf2xla/
Dfused_batchnorm_reserve_space_test.cc98 Tensor offset_data(DT_FLOAT, TensorShape({10})); in TEST() local
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_dnn.cc4422 const DeviceMemory<U>& scale_data, const DeviceMemory<U>& offset_data, in DoFusedBatchNormActivationInferenceImpl()
4498 const DeviceMemory<float>& offset_data, in DoFusedBatchNormActivationInference()
4514 const DeviceMemory<float>& offset_data, in DoFusedBatchNormActivationInference()
4531 const DeviceMemory<U>& scale_data, const DeviceMemory<U>& offset_data, in DoFusedBatchNormActivationForwardImpl()
4610 const DeviceMemory<float>& offset_data, double epsilon, in DoFusedBatchNormActivationForward()
4627 const DeviceMemory<float>& offset_data, double epsilon, in DoFusedBatchNormActivationForward()
4648 const DeviceMemory<U>& scale_data, const DeviceMemory<U>& offset_data, in DoFusedBatchNormActivationBackwardImpl()
4730 const DeviceMemory<float>& offset_data, in DoFusedBatchNormActivationBackward()
4753 const DeviceMemory<float>& offset_data, in DoFusedBatchNormActivationBackward()
/external/tensorflow/tensorflow/compiler/xla/tests/
Dbatch_normalization_test.cc632 std::unique_ptr<GlobalData> offset_data = in XLA_TEST_P() local
733 std::unique_ptr<GlobalData> offset_data = in XLA_TEST_P() local