| /external/tensorflow/tensorflow/core/kernels/ |
| adjust_contrast_op.h | 52 Eigen::Tensor<int, 4>::Dimensions reshape_dims; local
53 reshape_dims[0] = batch;
54 reshape_dims[1] = 1;
55 reshape_dims[2] = 1;
56 reshape_dims[3] = channels;
65 reshape_dims; local
66 reshape_dims.set(0, batch);
67 reshape_dims.set(3, channels);
74 .reshape(reshape_dims)
114 Eigen::Tensor<int, 4>::Dimensions reshape_dims; local
132 reshape_dims; local
[all...] |
| cwise_op_gpu_select.cu.cc | 69 Eigen::Tensor<int, 2>::Dimensions reshape_dims{{ batch, 1 }};
73 Eigen::IndexList<int, Eigen::type2index<1> > reshape_dims; local
74 reshape_dims.set(0, batch);
81 // .reshape(reshape_dims)
86 cond_vec.reshape(reshape_dims)
|
| gather_nd_op_cpu_impl.h | 107 Eigen::Tensor<Eigen::DenseIndex, 1>::Dimensions reshape_dims{{ 1 }};
110 Eigen::IndexList<Eigen::type2index<1> > reshape_dims; local
116 Tscratch.device(d) = Tscratch.reshape(reshape_dims)
|
| cwise_op_select.cc | 242 Eigen::Tensor<Eigen::DenseIndex, 2>::Dimensions reshape_dims{{batch, 1}};
246 Eigen::IndexList<Eigen::DenseIndex, Eigen::type2index<1> > reshape_dims; local
247 reshape_dims.set(0, batch);
251 cond_vec.reshape(reshape_dims)
|
| /external/tensorflow/tensorflow/contrib/lite/toco/graph_transformations/ |
| convert_expanddims_to_reshape.cc | 60 std::vector<int> reshape_dims(input_array.shape().dims());
62 axis = reshape_dims.size();
64 reshape_dims.insert(reshape_dims.begin() + axis, 1);
79 1, static_cast<int>(reshape_dims.size())};
83 shape_buffer.data = reshape_dims;
|
| convert_reorder_axes.cc | 40 std::vector<int> reshape_dims = {1, input_shape.dims(0), input_shape.dims(1), local
50 1, static_cast<int>(reshape_dims.size())};
54 reshape_buffer.data = reshape_dims;
|
| /external/eigen/unsupported/test/ |
| cxx11_tensor_padding.cpp | 63 Eigen::DSizes<ptrdiff_t, 2> reshape_dims; local
64 reshape_dims[0] = 12;
65 reshape_dims[1] = 84;
68 result = tensor.pad(paddings).reshape(reshape_dims);
|
| /external/tensorflow/tensorflow/tools/graph_transforms/ |
| quantize_nodes.cc | 743 NodeDef reshape_dims;
744 reshape_dims.set_op("Const");
745 reshape_dims.set_name(unique_input_name + "/reshape_dims");
746 AddNodeInput("^" + NodeNameFromInput(input_name), &reshape_dims);
747 SetNodeAttr("dtype", DT_INT32, &reshape_dims);
750 SetNodeTensorAttr<int32>("value", reshape_dims_tensor, &reshape_dims);
751 new_nodes->push_back(reshape_dims);
769 AddNodeInput(reshape_dims.name(), &reshape_node);
[all...] |
| /external/tensorflow/tensorflow/contrib/tensorrt/convert/ |
| convert_nodes.cc | 426 nvinfer1::Dims reshape_dims; local
427 reshape_dims.nbDims = dims.nbDims;
428 for (int32_t i = 0; i < reshape_dims.nbDims; ++i) {
429 reshape_dims.d[i] = 0;
430 reshape_dims.type[i] = dims.type[i];
432 layer->setReshapeDimensions(reshape_dims);
[all...] |
