Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Ddepthtospace_op.cc73 std::vector<int64> transpose_order; in Compile() local
76 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 …]
Dspacetodepth_op.cc73 std::vector<int64> transpose_order; in Compile() local
76 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 …]
/external/tensorflow/tensorflow/compiler/xla/client/lib/
Dtridiagonal.cc305 std::vector<int64> transpose_order(rank); in ThomasSolver() local
306 std::iota(transpose_order.begin(), transpose_order.end(), 0); in ThomasSolver()
307 transpose_order[rank - 2] = rank - 1; in ThomasSolver()
308 transpose_order[rank - 1] = rank - 2; in ThomasSolver()
310 rhs = Transpose(rhs, transpose_order); in ThomasSolver()
314 return Transpose(x, transpose_order); in ThomasSolver()
/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_nodes.cc2983 std::vector<int> transpose_order(input_dims.size()); in ConvertStridedSliceHelper() local
2987 transpose_order = {0, 2, 1, 3}; in ConvertStridedSliceHelper()
2990 transpose_order = {0, 3, 1, 2}; in ConvertStridedSliceHelper()
3005 tensor, transpose_order, &tensor, node_def, "for_pad", op_instance)); in ConvertStridedSliceHelper()