Searched refs:DimLookUp (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | space_to_batch_converter.cc | 199 int64 DimLookUp(absl::Span<const int64> permute_dims, int64 id) { in DimLookUp() function in xla::__anon6418faad0111::ConvolutionVisitor
832 DimLookUp(permute_dims_first_operand, old_batch_dim); in CanPropagate()
834 DimLookUp(permute_dims_first_operand, old_space_dim); in CanPropagate()
866 DimLookUp(permute_dims_second_operand, old_batch_dim); in CanPropagate()
894 DimLookUp(permute_dims_first_operand, dim_map_val_op_0.first); in CanPropagate()
896 DimLookUp(permute_dims_first_operand, dim_map_val_op_0.second); in CanPropagate()
899 DimLookUp(permute_dims_second_operand, dim_map_val_op_1.first); in CanPropagate()
901 DimLookUp(permute_dims_second_operand, dim_map_val_op_1.second); in CanPropagate()
967 DimLookUp(permute_dims_first_operand, old_batch_dim); in CanPropagate()
969 DimLookUp(permute_dims_first_operand, old_space_dim); in CanPropagate()
[all …]
|