Home
last modified time | relevance | path

Searched refs:permute_dims_first_operand (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dspace_to_batch_converter.cc1207 auto permute_dims_first_operand = in CanPropagate() local
1210 DimLookUp(permute_dims_first_operand, old_batch_dim); in CanPropagate()
1212 DimLookUp(permute_dims_first_operand, old_space_dim); in CanPropagate()
1281 auto permute_dims_first_operand = instr_to_dim_permute_map_[first_operand]; in CanPropagate() local
1286 DimLookUp(permute_dims_first_operand, in CanPropagate()
1290 DimLookUp(permute_dims_first_operand, in CanPropagate()
1357 auto permute_dims_first_operand = instr_to_dim_permute_map_[first_operand]; in CanPropagate() local
1362 if (permute_dims_first_operand != permute_dims_second_operand) { in CanPropagate()
1374 DimLookUp(permute_dims_first_operand, old_batch_dim); in CanPropagate()
1376 DimLookUp(permute_dims_first_operand, old_space_dim); in CanPropagate()
[all …]