Searched defs:activations_batch_dim (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | space_to_batch_converter.cc | 323 int64 activations_batch_dim = dim_numbers.input_batch_dimension(); in IsConvSuitableForSpaceToBatch() local
345 int64 activations_batch_dim, int64 old_batch_size, int64 low_padding, in HaloDuplicateWithSlice()
480 int64& spatial_dimension_to_split, int64& activations_batch_dim, in BringSpaceNextToBatch()
1861 int64 activations_batch_dim = in PropagateOnConv() local
2039 int64 activations_batch_dim, int64 high_padding, int64 low_padding, in SplitSpaceHelper()
2094 int64& activations_batch_dim, int64 high_padding, in SplitSpace()
2280 int64 activations_batch_dim = in PropagateOnBackpropFilterConv() local
2663 int64 activations_batch_dim = dim_numbers.input_batch_dimension(); in PerformSpaceToBatchOnConvolution() local
|