Searched defs:offset_data (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | fused_batch_norm_ex_op_test.cc | 138 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()
|
| D | conv_ops_test.cc | 668 const Tensor& scale_data, const Tensor& offset_data, in RunConv2DWithBatchNorm()
697 const Tensor& scale_data, const Tensor& offset_data, in RunConv2DWithBatchNormAndActivation()
954 const Tensor& scale_data, const Tensor& offset_data,
977 const Tensor& scale_data, const Tensor& offset_data,
987 const Tensor& scale_data, const Tensor& offset_data,
988 const Tensor& mean_data, const Tensor& variance_data, Tensor* out) { in __anone7ca09510902()
|
| D | fused_eigen_output_kernels.h | 149 const T* offset_data = nullptr; member
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | fused_batchnorm_reserve_space_test.cc | 99 Tensor offset_data(DT_FLOAT, TensorShape({10})); in TEST() local
|
| /external/tensorflow/tensorflow/stream_executor/ |
| D | dnn.h | 2532 const DeviceMemory<float>& offset_data, in DoFusedBatchNormActivationInference()
2545 const DeviceMemory<float>& offset_data, in DoFusedBatchNormActivationInference()
2586 const DeviceMemory<float>& offset_data, double epsilon, in DoFusedBatchNormActivationForward()
2599 const DeviceMemory<float>& offset_data, double epsilon, in DoFusedBatchNormActivationForward()
2646 const DeviceMemory<float>& offset_data, in DoFusedBatchNormActivationBackward()
2664 const DeviceMemory<float>& offset_data, in DoFusedBatchNormActivationBackward()
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_dnn.cc | 4730 const DeviceMemory<U>& scale_data, const DeviceMemory<U>& offset_data, in DoFusedBatchNormActivationInferenceImpl()
4806 const DeviceMemory<float>& offset_data, in DoFusedBatchNormActivationInference()
4822 const DeviceMemory<float>& offset_data, in DoFusedBatchNormActivationInference()
4839 const DeviceMemory<U>& scale_data, const DeviceMemory<U>& offset_data, in DoFusedBatchNormActivationForwardImpl()
4918 const DeviceMemory<float>& offset_data, double epsilon, in DoFusedBatchNormActivationForward()
4935 const DeviceMemory<float>& offset_data, double epsilon, in DoFusedBatchNormActivationForward()
4956 const DeviceMemory<U>& scale_data, const DeviceMemory<U>& offset_data, in DoFusedBatchNormActivationBackwardImpl()
5038 const DeviceMemory<float>& offset_data, in DoFusedBatchNormActivationBackward()
5061 const DeviceMemory<float>& offset_data, in DoFusedBatchNormActivationBackward()
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | batch_normalization_test.cc | 632 std::unique_ptr<GlobalData> offset_data = in XLA_TEST_P() local
733 std::unique_ptr<GlobalData> offset_data = in XLA_TEST_P() local
|
| /external/llvm-project/lldb/source/Plugins/Process/gdb-remote/ |
| D | GDBRemoteCommunicationServerLLGS.cpp | 2984 StringExtractor offset_data(fields[4]); in Handle_qXfer() local
|