Searched refs:SplitSpace (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | space_to_batch_converter.cc | 116 StatusOr<std::pair<HloInstruction*, std::vector<int64>>> SplitSpace(
2091 ConvolutionVisitor::SplitSpace(HloInstruction* activations, in SplitSpace() function in xla::__anon6418faad0111::ConvolutionVisitor
2203 SplitSpace(activations_old, tmp_dim_numbers, old_space_dim, in PropagateOnBackpropFilterConv()
2247 auto retval, SplitSpace(kernel_old, tmp_dim_numbers, kernel_space_dim, in PropagateOnBackpropFilterConv()
2719 SplitSpace(activations, dim_numbers, spatial_dimension_to_split, in PerformSpaceToBatchOnConvolution()
|