Searched refs:dimensions_without_feature (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | batchnorm_expander.cc | 223 std::vector<int64> dimensions_without_feature; in HandleBatchNormTraining() local
227 dimensions_without_feature.push_back(i); in HandleBatchNormTraining()
248 dimensions_without_feature, in HandleBatchNormTraining()
253 feature_shape, operand_squared, zero, dimensions_without_feature, in HandleBatchNormTraining()
351 std::vector<int64> dimensions_without_feature; in HandleBatchNormInference() local
355 dimensions_without_feature.push_back(i); in HandleBatchNormInference()
488 std::vector<int64> dimensions_without_feature; in HandleBatchNormGrad() local
492 dimensions_without_feature.push_back(i); in HandleBatchNormGrad()
531 dimensions_without_feature, add_reduce_computation)); in HandleBatchNormGrad()
535 feature_shape, grad_output, zero, dimensions_without_feature, in HandleBatchNormGrad()
|