| /external/tensorflow/tensorflow/core/kernels/ |
| D | adjust_contrast_op.h | 52 Eigen::Tensor<int, 4>::Dimensions reshape_dims; in operator() local
53 reshape_dims[0] = batch; in operator()
54 reshape_dims[1] = 1; in operator()
55 reshape_dims[2] = 1; in operator()
56 reshape_dims[3] = channels; in operator()
65 reshape_dims; in operator() local
66 reshape_dims.set(0, batch); in operator()
67 reshape_dims.set(3, channels); in operator()
74 .reshape(reshape_dims) in operator()
114 Eigen::Tensor<int, 4>::Dimensions reshape_dims; in operator() local
[all …]
|
| D | cwise_op_gpu_select.cu.cc | 69 Eigen::Tensor<int, 2>::Dimensions reshape_dims{{ batch, 1 }}; in operator ()() local
73 Eigen::IndexList<int, Eigen::type2index<1> > reshape_dims; in operator ()() local
74 reshape_dims.set(0, batch); in operator ()()
86 cond_vec.reshape(reshape_dims) in operator ()()
|
| D | gather_nd_op_cpu_impl.h | 107 Eigen::Tensor<Eigen::DenseIndex, 1>::Dimensions reshape_dims{{ 1 }};
110 Eigen::IndexList<Eigen::type2index<1> > reshape_dims;
130 Tscratch.device(d) = Tscratch.reshape(reshape_dims)
|
| D | cwise_op_select.cc | 270 Eigen::Tensor<Eigen::DenseIndex, 2>::Dimensions reshape_dims{{batch, 1}}; in operator ()() local
274 Eigen::IndexList<Eigen::DenseIndex, Eigen::type2index<1> > reshape_dims; in operator ()() local
275 reshape_dims.set(0, batch); in operator ()()
279 cond_vec.reshape(reshape_dims) in operator ()()
|
| D | eigen_spatial_convolutions_test.cc | 1490 NewDimension reshape_dims; in PackRhsHelper() local
1491 reshape_dims[0] = input_depth * filter_rows * filter_cols; // patch size in PackRhsHelper()
1492 reshape_dims[1] = output_rows * output_cols * input_batches; // num_patches in PackRhsHelper()
1496 image_patch_op, reshape_dims); in PackRhsHelper()
1626 NewDimension reshape_dims; in PackLhsHelper() local
1627 reshape_dims[0] = filter_count; in PackLhsHelper()
1628 reshape_dims[1] = input_depth * filter_rows * filter_cols; in PackLhsHelper()
1662 TensorReshapingOp<NewDimension, ArgType>(tensor_map, reshape_dims); in PackLhsHelper()
|
| /external/tensorflow/tensorflow/lite/toco/graph_transformations/ |
| D | convert_expanddims_to_reshape.cc | 58 std::vector<int> reshape_dims(input_array.shape().dims()); in Run() local
60 axis = reshape_dims.size(); in Run()
62 reshape_dims.insert(reshape_dims.begin() + axis, 1); in Run()
77 1, static_cast<int>(reshape_dims.size())}; in Run()
81 shape_buffer.data = reshape_dims; in Run()
|
| D | convert_reorder_axes.cc | 40 std::vector<int> reshape_dims = {1, input_shape.dims(0), input_shape.dims(1), in CreateReshapeFromReorderAxes() local
50 1, static_cast<int>(reshape_dims.size())}; in CreateReshapeFromReorderAxes()
54 reshape_buffer.data = reshape_dims; in CreateReshapeFromReorderAxes()
|
| /external/eigen/unsupported/test/ |
| D | cxx11_tensor_padding.cpp | 63 Eigen::DSizes<ptrdiff_t, 2> reshape_dims; in test_padded_expr() local
64 reshape_dims[0] = 12; in test_padded_expr()
65 reshape_dims[1] = 84; in test_padded_expr()
68 result = tensor.pad(paddings).reshape(reshape_dims); in test_padded_expr()
|
| /external/tensorflow/tensorflow/tools/graph_transforms/ |
| D | quantize_nodes.cc | 742 NodeDef reshape_dims; in QuantizeNodes() local
743 reshape_dims.set_op("Const"); in QuantizeNodes()
744 reshape_dims.set_name(unique_input_name + "/reshape_dims"); in QuantizeNodes()
745 AddNodeInput("^" + NodeNameFromInput(input_name), &reshape_dims); in QuantizeNodes()
746 SetNodeAttr("dtype", DT_INT32, &reshape_dims); in QuantizeNodes()
749 SetNodeTensorAttr<int32>("value", reshape_dims_tensor, &reshape_dims); in QuantizeNodes()
750 new_nodes->push_back(reshape_dims); in QuantizeNodes()
768 AddNodeInput(reshape_dims.name(), &reshape_node); in QuantizeNodes()
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_nodes.cc | 1180 nvinfer1::Dims reshape_dims; in TransposeTensor() local
1181 reshape_dims.nbDims = dims.nbDims; in TransposeTensor()
1182 for (int32_t i = 0; i < reshape_dims.nbDims; ++i) { in TransposeTensor()
1183 reshape_dims.d[i] = 0; in TransposeTensor()
1185 reshape_dims.type[i] = dims.type[i]; in TransposeTensor()
1187 layer->setReshapeDimensions(reshape_dims); in TransposeTensor()
2131 nvinfer1::Dims reshape_dims; in ConvertReshape() local
2132 reshape_dims.nbDims = weights.count() - 1; in ConvertReshape()
2134 reshape_dims.d[i - 1] = weights_ptr[i]; in ConvertReshape()
2143 TrtDimsNumElements(reshape_dims) != TrtDimsNumElements(input_dims)) { in ConvertReshape()
[all …]
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | resampler_ops.cc | 264 std::vector<int64> reshape_dims(warp_shape.dims()); in CalculateGradData() local
265 std::iota(reshape_dims.begin(), reshape_dims.end(), 0); in CalculateGradData()
267 auto reshaped_weights = xla::Reshape(weights, /*dimensions=*/reshape_dims, in CalculateGradData()
|