Searched defs:offset_data (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | fused_batchnorm_reserve_space_test.cc | 85 Tensor offset_data(DT_FLOAT, TensorShape({10})); in TEST() local
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | conv_ops_fused_impl.h | 224 const T* offset_data; member
|
| D | conv_ops_test.cc | 651 const Tensor& offset_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/stream_executor/rocm/ |
| D | rocm_dnn.cc | 4134 const DeviceMemory<U>& scale_data, const DeviceMemory<U>& offset_data, in DoFusedBatchNormActivationInferenceImpl()
4210 const DeviceMemory<float>& offset_data, in DoFusedBatchNormActivationInference()
4226 const DeviceMemory<float>& offset_data, in DoFusedBatchNormActivationInference()
4243 const DeviceMemory<U>& scale_data, const DeviceMemory<U>& offset_data, in DoFusedBatchNormActivationForwardImpl()
4322 const DeviceMemory<float>& offset_data, double epsilon, in DoFusedBatchNormActivationForward()
4339 const DeviceMemory<float>& offset_data, double epsilon, in DoFusedBatchNormActivationForward()
4360 const DeviceMemory<U>& scale_data, const DeviceMemory<U>& offset_data, in DoFusedBatchNormActivationBackwardImpl()
4442 const DeviceMemory<float>& offset_data, in DoFusedBatchNormActivationBackward()
4465 const DeviceMemory<float>& offset_data, in DoFusedBatchNormActivationBackward()
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | batch_normalization_test.cc | 534 std::unique_ptr<GlobalData> offset_data = in XLA_TEST_P() local
635 std::unique_ptr<GlobalData> offset_data = in XLA_TEST_P() local
|