Searched refs:permute_dims_kernel (Results 1 – 1 of 1) sorted by relevance
2179 std::vector<int64> permute_dims_kernel, permute_dims; in PropagateOnBackpropFilterConv() local2183 permute_dims_kernel = instr_to_dim_permute_map_[kernel_new]; in PropagateOnBackpropFilterConv()2192 DimLookUp(permute_dims_kernel, kernel_space_dim); in PropagateOnBackpropFilterConv()2274 permute_dims_kernel = instr_to_dim_permute_map_[kernel_new]; in PropagateOnBackpropFilterConv()2292 i, DimLookUp(permute_dims_kernel, in PropagateOnBackpropFilterConv()2313 permute_dims_kernel, original_conv_dims.kernel_input_feature_dimension()); in PropagateOnBackpropFilterConv()2316 DimLookUp(permute_dims_kernel, in PropagateOnBackpropFilterConv()