| /external/tensorflow/tensorflow/core/kernels/image/ |
| 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 …]
|
| /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
59 int original_dims_num = reshape_dims.size(); in Run()
66 axis = reshape_dims.size() + 1 + axis; in Run()
68 reshape_dims.insert(reshape_dims.begin() + axis, 1); in Run()
84 1, static_cast<int>(reshape_dims.size())}; in Run()
88 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/compiler/xla/service/gpu/ |
| D | gpu_conv_rewriter.cc | 87 std::vector<int64> reshape_dims = SpanToVector(lhs->shape().dimensions()); in ConvertBatchGroupedToFeatureGroupedConvolution() local
88 reshape_dims[input_batch_dimension] = in ConvertBatchGroupedToFeatureGroupedConvolution()
89 reshape_dims[input_batch_dimension] / num_groups; in ConvertBatchGroupedToFeatureGroupedConvolution()
90 reshape_dims.insert(reshape_dims.begin() + input_batch_dimension, num_groups); in ConvertBatchGroupedToFeatureGroupedConvolution()
92 ShapeUtil::MakeShape(lhs->shape().element_type(), reshape_dims), lhs)); in ConvertBatchGroupedToFeatureGroupedConvolution()
567 std::vector<int64> reshape_dims = SpanToVector(rhs->shape().dimensions()); in MatchBackwardInput() local
572 reshape_dims[input_feature_dimension] = in MatchBackwardInput()
573 reshape_dims[input_feature_dimension] / num_groups; in MatchBackwardInput()
574 reshape_dims.insert(reshape_dims.begin() + input_feature_dimension, in MatchBackwardInput()
579 ShapeUtil::MakeShape(rhs->shape().element_type(), reshape_dims), rhs)); in MatchBackwardInput()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | cwise_op_gpu_select.cu.cc | 85 Eigen::Tensor<int, 2>::Dimensions reshape_dims{{ batch, 1 }}; in operator ()() local
89 Eigen::IndexList<int, Eigen::type2index<1> > reshape_dims; in operator ()() local
90 reshape_dims.set(0, batch); in operator ()()
102 cond_vec.reshape(reshape_dims) in operator ()()
|
| D | cwise_op_select.cc | 342 Eigen::Tensor<Eigen::DenseIndex, 2>::Dimensions reshape_dims{{batch, 1}}; in operator ()() local
346 Eigen::IndexList<Eigen::DenseIndex, Eigen::type2index<1> > reshape_dims; in operator ()() local
347 reshape_dims.set(0, batch); in operator ()()
351 cond_vec.reshape(reshape_dims) in operator ()()
|
| D | eigen_spatial_convolutions_test.cc | 1511 NewDimension reshape_dims; in PackRhsHelper() local
1512 reshape_dims[0] = input_depth * filter_rows * filter_cols; // patch size in PackRhsHelper()
1513 reshape_dims[1] = output_rows * output_cols * input_batches; // num_patches in PackRhsHelper()
1517 image_patch_op, reshape_dims); in PackRhsHelper()
1649 NewDimension reshape_dims; in PackLhsHelper() local
1650 reshape_dims[0] = filter_count; in PackLhsHelper()
1651 reshape_dims[1] = input_depth * filter_rows * filter_cols; in PackLhsHelper()
1689 TensorReshapingOp<NewDimension, ArgType>(tensor_map, reshape_dims); in PackLhsHelper()
|
| /external/tensorflow/tensorflow/tools/graph_transforms/ |
| D | quantize_nodes.cc | 747 NodeDef reshape_dims; in QuantizeNodes() local
748 reshape_dims.set_op("Const"); in QuantizeNodes()
749 reshape_dims.set_name(unique_input_name + "/reshape_dims"); in QuantizeNodes()
750 AddNodeInput("^" + NodeNameFromInput(input_name), &reshape_dims); in QuantizeNodes()
751 SetNodeAttr("dtype", DT_INT32, &reshape_dims); in QuantizeNodes()
754 SetNodeTensorAttr<int32>("value", reshape_dims_tensor, &reshape_dims); in QuantizeNodes()
755 new_nodes->push_back(reshape_dims); in QuantizeNodes()
773 AddNodeInput(reshape_dims.name(), &reshape_node); in QuantizeNodes()
|
| /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()
|
| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| D | prng.cc | 228 std::vector<int64> reshape_dims(original_shape.dimensions().begin(), in CombineShapePair() local
230 reshape_dims[shape_pair.split_dim] = in CombineShapePair()
232 result = Reshape(result, reshape_dims); in CombineShapePair()
233 if (reshape_dims[shape_pair.split_dim] != pre_split_size) { in CombineShapePair()
|
| /external/tensorflow/tensorflow/compiler/mlir/tosa/transforms/ |
| D | legalize_common.cc | 1265 SmallVector<int64_t, 4> reshape_dims; in convertExpandDimsOp() local
1269 reshape_dims.emplace_back(input_shape[i]); in convertExpandDimsOp()
1271 reshape_dims.emplace_back(1); in convertExpandDimsOp()
1275 reshape_dims.emplace_back(1); in convertExpandDimsOp()
1277 reshape_dims.emplace_back(input_shape[i]); in convertExpandDimsOp()
1281 ArrayAttr shape_attr = rewriter.getI64ArrayAttr(reshape_dims); in convertExpandDimsOp()
1312 SmallVector<int64_t, 8> reshape_dims; in convertSqueezeOp() local
1317 reshape_dims.emplace_back(input_shape[i]); in convertSqueezeOp()
1336 reshape_dims.emplace_back(input_shape[i]); in convertSqueezeOp()
1341 ArrayAttr shape_attr = rewriter.getI64ArrayAttr(reshape_dims); in convertSqueezeOp()
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_nodes.cc | 1616 nvinfer1::Dims reshape_dims; in TransposeTensor() local
1617 reshape_dims.nbDims = dims.nbDims; in TransposeTensor()
1618 for (int32_t i = 0; i < reshape_dims.nbDims; ++i) { in TransposeTensor()
1619 reshape_dims.d[i] = 0; in TransposeTensor()
1621 reshape_dims.type[i] = dims.type[i]; in TransposeTensor()
1623 layer->setReshapeDimensions(reshape_dims); in TransposeTensor()
2922 nvinfer1::Dims reshape_dims; in ConvertStridedSliceHelper() local
2931 TF_RETURN_IF_ERROR(TensorShapeArrayToTrtDims(input_dims, &reshape_dims, in ConvertStridedSliceHelper()
3000 params->converter, input, reshape_dims, /*validation_only=*/false, in ConvertStridedSliceHelper()
4113 nvinfer1::Dims reshape_dims; in ConvertBiasAddInt8WithoutCalibration() local
[all …]
|
| D | convert_nodes_test.cc | 937 const std::vector<int>& input_dims, const std::vector<int>& reshape_dims, in TestPrepareTensorForShape() argument
955 PrepareTensorForShape(converter, input, GetTestDims(reshape_dims), in TestPrepareTensorForShape()
|
| /external/tensorflow/tensorflow/compiler/mlir/tosa/g3doc/ |
| D | legalization.md | 412 vector<size_t> reshape_dims
418 reshape_dims.push_back(%input.shape[i])
423 reshape_dims.push_back(1)
425 reshape_dims.push_back(%input.shape[i])
429 …%output = tosa.RESHAPE(%input) {new_shape=reshape_dims} (tensor<%input.type>) -> tensor<%output.ty…
886 vector <size_t> reshape_dims
892 reshape_dims.push_back(%input_shape[i])
899 reshape_dims.push_back(%input_shape[i])
904 …%output = tosa.RESHAPE(%input) {new_shape=reshape_dims} (tensor<%input.type>) -> tensor<reshape_di…
|