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 | 446 int64_t activations_batch_dim = dim_numbers.input_batch_dimension(); in IsConvSuitableForSpaceToBatch() local
512 int64_t activations_batch_dim, int64_t low_padding, int64_t halo_size, in HaloDuplicateWithSlice()
641 int64_t& activations_batch_dim, in BringSpaceNextToBatch()
2551 int64_t activations_batch_dim = in PropagateOnConv() local
2815 int64_t activations_batch_dim, int64_t old_batch_size) { in TransposeAndMergeBatch()
2857 int64_t activations_batch_dim, int64_t spatial_split_size, in PerformSplitSpace()
2903 int64_t activations_batch_dim, int64_t high_padding, int64_t low_padding, in PadAndSplitSpace()
2937 int64_t& activations_batch_dim, int64_t high_padding, int64_t low_padding, in SplitSpace()
3136 int64_t activations_batch_dim = in PropagateOnBackpropFilterConv() local
3641 int64_t activations_batch_dim = dim_numbers.input_batch_dimension(); in PerformSpaceToBatchOnConvolution() local
|