Searched refs:transpose_order (Results 1 – 3 of 3) sorted by relevance
73 std::vector<int64> transpose_order; in Compile() local76 transpose_order.reserve(input_rank); in Compile()90 transpose_order.push_back(0); in Compile()92 transpose_order.push_back(i + 1); in Compile()93 transpose_order.push_back(i + 1 + num_spatial_dims); in Compile()95 transpose_order.push_back(feature_dim + num_spatial_dims); in Compile()115 transpose_order.push_back(0); in Compile()116 transpose_order.push_back(1 + num_spatial_dims); in Compile()118 transpose_order.push_back(2 + num_spatial_dims + i); in Compile()119 transpose_order.push_back(1 + i); in Compile()[all …]
73 std::vector<int64> transpose_order; in Compile() local76 transpose_order.reserve(input_rank); in Compile()95 transpose_order.push_back(0); in Compile()97 transpose_order.push_back(i * 2 + 1); in Compile()100 transpose_order.push_back(i * 2 + 2); in Compile()102 transpose_order.push_back(feature_dim + num_spatial_dims); in Compile()127 transpose_order.push_back(0); in Compile()129 transpose_order.push_back(i * 2 + 3); in Compile()131 transpose_order.push_back(feature_dim); in Compile()133 transpose_order.push_back(i * 2 + 2); in Compile()[all …]
2421 std::vector<int> transpose_order(input_dims.size()); in ConvertStridedSliceHelper() local2425 transpose_order = {0, 2, 1, 3}; in ConvertStridedSliceHelper()2428 transpose_order = {0, 3, 1, 2}; in ConvertStridedSliceHelper()2445 tensor, transpose_order, &output_tensor)); in ConvertStridedSliceHelper()