| /external/tensorflow/tensorflow/core/kernels/ |
| transpose_util_test.cc | 30 internal::TransposeDimsVec new_dims; local
31 internal::ReduceTransposeDimensions(shape, perm, &new_perm, &new_dims);
34 gtl::ArraySlice<int64> computed_dims(new_dims);
|
| transpose_functor_gpu.cu.cc | 101 TransposeDimsVec new_dims; local
102 ReduceTransposeDimensions(in.shape(), perm, &new_perm, &new_dims);
105 int dims = new_dims.size();
114 new_dims.insert(new_dims.begin(), 1);
117 d, in_data, new_dims, out_data);
125 d, in_data, new_dims, out_data);
130 d, in_data, new_dims, out_data);
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| diag_op.cc | 110 std::vector<int64> new_dims(dims.size() * 2);
111 std::copy(dims.begin(), dims.end(), new_dims.begin());
112 std::copy(dims.begin(), dims.end(), new_dims.begin() + dims.size());
113 diag = xla::Reshape(diag, new_dims);
140 std::vector<int64> new_dims; variable
148 new_dims.push_back(dims[i]);
155 new_dims);
|
| /external/tensorflow/tensorflow/lite/toco/graph_transformations/ |
| resolve_constant_unary.cc | 85 std::vector<int> new_dims; local
88 new_dims.push_back(output_shape.dims(i));
91 output_shape.mutable_dims()->swap(new_dims);
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| indexed_array_analysis.cc | 458 DimensionVector new_dims; local
459 absl::c_copy_if(shape.dimensions(), std::back_inserter(new_dims),
461 return ShapeUtil::MakeShape(shape.element_type(), new_dims);
[all...] |
| /external/tensorflow/tensorflow/c/ |
| c_api_test.cc | 1485 int64_t new_dims[] = {3, 2}; local
[all...] |
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| constant_folding.cc | 1535 PartialTensorShape new_dims; local
[all...] |
