HomeSort by relevance Sort by last modified time
    Searched refs:transpose_order (Results 1 - 2 of 2) sorted by null

  /external/tensorflow/tensorflow/compiler/tf2xla/kernels/
depthtospace_op.cc 60 std::vector<int64> transpose_order; variable
63 transpose_order.reserve(input_rank);
77 transpose_order.push_back(0);
79 transpose_order.push_back(i + 1);
80 transpose_order.push_back(i + 1 + num_spatial_dims);
82 transpose_order.push_back(feature_dim + num_spatial_dims);
102 transpose_order.push_back(0);
103 transpose_order.push_back(1 + num_spatial_dims);
105 transpose_order.push_back(2 + num_spatial_dims + i);
106 transpose_order.push_back(1 + i)
    [all...]
spacetodepth_op.cc 60 std::vector<int64> transpose_order; variable
63 transpose_order.reserve(input_rank);
82 transpose_order.push_back(0);
84 transpose_order.push_back(i * 2 + 1);
87 transpose_order.push_back(i * 2 + 2);
89 transpose_order.push_back(feature_dim + num_spatial_dims);
114 transpose_order.push_back(0);
116 transpose_order.push_back(i * 2 + 3);
118 transpose_order.push_back(feature_dim);
120 transpose_order.push_back(i * 2 + 2)
    [all...]

Completed in 178 milliseconds