Searched refs:trans_dims (Results 1 – 1 of 1) sorted by relevance
1893 std::vector<int64> trans_dims = retval.transpose_dims; in PropagateOnConv() local1894 CHECK(!trans_dims.empty()); in PropagateOnConv()2557 std::vector<int64> trans_dims(convolution->shape().dimensions_size()); in PropagateOnBackpropFilterConv() local2558 absl::c_iota(trans_dims, 0); in PropagateOnBackpropFilterConv()2559 instr_to_dim_permute_map_[new_conv] = trans_dims; in PropagateOnBackpropFilterConv()