Searched defs:mean_data (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/tools/graph_transforms/ |
| D | fold_old_batch_norms_test.cc | 60 Tensor mean_data(DT_FLOAT, TensorShape({2})); in TestFoldOldBatchNorms() local
144 Tensor mean_data(DT_FLOAT, TensorShape({4})); in TestFoldOldBatchNormsAfterDepthwiseConv2dNative() local
222 Tensor mean_data(DT_FLOAT, TensorShape({2})); in TestFoldFusedBatchNorms() local
299 Tensor mean_data(DT_FLOAT, TensorShape({4})); in TestFoldFusedBatchNormsAfterDepthwiseConv2dNative() local
400 Tensor mean_data(DT_FLOAT, TensorShape({2})); in TestFoldFusedBatchNormsWithConcat() local
492 Tensor mean_data(DT_FLOAT, TensorShape({2})); in TestFoldFusedBatchNormsWithBatchToSpace() local
|
| /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()
446 const Tensor& mean_data, const Tensor& var_data, in VerifyFusedBatchNormEx()
448 FusedBatchNormGradOutputs* bwd) { in VerifyFusedBatchNormEx()
|
| D | conv_ops_test.cc | 665 const Tensor& mean_data, const Tensor& variance_data, in RunConv2DWithBatchNorm()
694 const Tensor& mean_data, const Tensor& variance_data, in RunConv2DWithBatchNormAndActivation()
938 const Tensor& mean_data, const Tensor& variance_data, Tensor* out) { in __anon10aa6af50602()
948 const Tensor& mean_data, const Tensor& variance_data, Tensor* out) { in __anon10aa6af50702()
971 const Tensor& mean_data, const Tensor& variance_data, Tensor* out) { in __anon10aa6af50802()
981 const Tensor& mean_data, const Tensor& variance_data, Tensor* out) { in __anon10aa6af50902()
|
| D | mkl_fused_batch_norm_op.cc | 69 U* mean_data, U* variance_data) { in Execute()
320 void Execute(const T* src_data, const U* mean_data, const U* variance_data, in Execute()
1033 U* mean_data = in Compute() local
|
| D | adjust_contrast_op.cc | 218 auto mean_data = mean_values.tensor<float, 2>(); in DoCompute() local
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | fused_batchnorm_reserve_space_test.cc | 103 Tensor mean_data(DT_FLOAT, TensorShape({0})); in TEST() local
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | batch_normalization_test.cc | 735 std::unique_ptr<GlobalData> mean_data = in XLA_TEST_P() local
918 std::unique_ptr<GlobalData> mean_data = in XLA_TEST_P() local
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_dnn.cc | 4423 const DeviceMemory<U>& mean_data, const DeviceMemory<U>& variance_data, in DoFusedBatchNormActivationInferenceImpl()
4499 const DeviceMemory<float>& mean_data, in DoFusedBatchNormActivationInference()
4515 const DeviceMemory<float>& mean_data, in DoFusedBatchNormActivationInference()
|