Searched refs:spatial_dimension_to_split (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | space_to_batch_converter.cc | 71 int64 spatial_dimension_to_split, inherent_low_padding, member
118 int64& spatial_dimension_to_split, int64& activations_batch_dim,
125 HloInstruction* activations, int64 spatial_dimension_to_split,
163 int64& spatial_dimension_to_split, int64& activations_batch_dim,
177 HloInstruction* activations, int64 spatial_dimension_to_split,
344 HloInstruction* activations, int64 spatial_dimension_to_split, in HaloDuplicateWithSlice() argument
362 spatial_dimension_to_split++; in HaloDuplicateWithSlice()
368 activations->shape().dimensions(spatial_dimension_to_split); in HaloDuplicateWithSlice()
391 start_indices[spatial_dimension_to_split] = in HaloDuplicateWithSlice()
394 end_indices[spatial_dimension_to_split] = spatial_split_size; in HaloDuplicateWithSlice()
[all …]
|