Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dspace_to_batch_converter.cc763 auto dim_map_val_op_0 = instr_to_dim_map_[consumer->mutable_operand(0)]; in CanPropagate() local
767 get_chosen_spatial_dim(consumer)) != dim_map_val_op_0.second) { in CanPropagate()
773 dim_map_val_op_0.first) { in CanPropagate()
825 auto dim_map_val_op_0 = instr_to_dim_map_[activations]; in CanPropagate() local
826 const int64 old_batch_dim = dim_map_val_op_0.first; in CanPropagate()
827 const int64 old_space_dim = dim_map_val_op_0.second; in CanPropagate()
885 auto dim_map_val_op_0 = instr_to_dim_map_[activations]; in CanPropagate() local
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()
949 auto dim_map_val_op_0 = instr_to_dim_map_[consumer->mutable_operand(0)]; in CanPropagate() local
[all …]