Searched refs:reversed_transpose_dims (Results 1 – 1 of 1) sorted by relevance
2118 std::vector<int64> reversed_transpose_dims(prod_transpose_dims.size()); in PropagateOnConstant() local2120 reversed_transpose_dims[i] = ReverseDimLookUp(prod_transpose_dims, i); in PropagateOnConstant()2124 MakeTransposeHlo(consumer, reversed_transpose_dims)); in PropagateOnConstant()2210 std::vector<int64> reversed_transpose_dims(retval.second.size()); in PropagateOnBackpropFilterConv() local2212 reversed_transpose_dims[i] = ReverseDimLookUp(retval.second, i); in PropagateOnBackpropFilterConv()2214 instr_to_dim_permute_map_[retval.first] = reversed_transpose_dims; in PropagateOnBackpropFilterConv()2255 std::vector<int64> reversed_transpose_dims(retval.second.size()); in PropagateOnBackpropFilterConv() local2257 reversed_transpose_dims[i] = ReverseDimLookUp(retval.second, i); in PropagateOnBackpropFilterConv()2259 instr_to_dim_permute_map_[retval.first] = reversed_transpose_dims; in PropagateOnBackpropFilterConv()