Searched refs:activations_batch_dim (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | space_to_batch_converter.cc | 118 int64& spatial_dimension_to_split, int64& activations_batch_dim,
126 int64 activations_batch_dim, int64 high_padding, int64 low_padding,
163 int64& spatial_dimension_to_split, int64& activations_batch_dim,
178 int64 activations_batch_dim, int64 old_batch_size, int64 low_padding,
323 int64 activations_batch_dim = dim_numbers.input_batch_dimension(); in IsConvSuitableForSpaceToBatch() local
326 convolution->operand(0)->shape().dimensions(activations_batch_dim); in IsConvSuitableForSpaceToBatch()
345 int64 activations_batch_dim, int64 old_batch_size, int64 low_padding, in HaloDuplicateWithSlice() argument
349 activations->shape().dimensions(activations_batch_dim) / kNumSplits; in HaloDuplicateWithSlice()
354 new_dimensions[activations_batch_dim] = kNumSplits; in HaloDuplicateWithSlice()
355 new_dimensions.insert(new_dimensions.begin() + activations_batch_dim, in HaloDuplicateWithSlice()
[all …]
|