| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | diag_op.cc | 110 std::vector<int64> new_dims(dims.size() * 2); in Compile() local
111 std::copy(dims.begin(), dims.end(), new_dims.begin()); in Compile()
112 std::copy(dims.begin(), dims.end(), new_dims.begin() + dims.size()); in Compile()
113 diag = xla::Reshape(diag, new_dims); in Compile()
140 std::vector<int64> new_dims; in Compile() local
148 new_dims.push_back(dims[i]); in Compile()
155 new_dims); in Compile()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | transpose_functor_gpu.cu.cc | 101 TransposeDimsVec new_dims; in run() local
102 ReduceTransposeDimensions(in.shape(), perm, &new_perm, &new_dims); in run()
105 int dims = new_dims.size(); in run()
114 new_dims.insert(new_dims.begin(), 1); in run()
117 d, in_data, new_dims, out_data); in run()
125 d, in_data, new_dims, out_data); in run()
130 d, in_data, new_dims, out_data); in run()
|
| D | transpose_util_test.cc | 30 internal::TransposeDimsVec new_dims; in TestDimensionReduction() local
31 internal::ReduceTransposeDimensions(shape, perm, &new_perm, &new_dims); in TestDimensionReduction()
34 gtl::ArraySlice<int64> computed_dims(new_dims); in TestDimensionReduction()
|
| D | transpose_functor.h | 81 TransposeDimsVec* new_dims) { in ReduceTransposeDimensions() argument
87 (*new_dims)[0] = shape.dim_size(0); in ReduceTransposeDimensions()
112 new_dims->resize(dim_idx + 1); in ReduceTransposeDimensions()
118 (*new_dims)[dim_idx] = combined_dims[new_perm_idx]; in ReduceTransposeDimensions()
|
| /external/tensorflow/tensorflow/lite/toco/graph_transformations/ |
| D | resolve_constant_unary.cc | 85 std::vector<int> new_dims; in ReduceGeneric() local
88 new_dims.push_back(output_shape.dims(i)); in ReduceGeneric()
91 output_shape.mutable_dims()->swap(new_dims); in ReduceGeneric()
|
| /external/tensorflow/tensorflow/python/framework/ |
| D | tensor_shape.py | 896 new_dims = []
898 new_dims.append(dim.merge_with(other[i]))
899 return TensorShape(new_dims)
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | xla_op_kernel.cc | 139 int index, absl::Span<const int64> new_dims, in ConstantInputReshaped() argument
166 if (!temp.CopyFrom(*constant, TensorShape(new_dims))) { in ConstantInputReshaped()
171 TensorShape(new_dims).DebugString()); in ConstantInputReshaped()
|
| D | xla_op_kernel.h | 274 Status ConstantInputReshaped(int index, absl::Span<const int64> new_dims,
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | indexed_array_analysis.cc | 458 DimensionVector new_dims; in StripDegenerateDimensions() local
459 absl::c_copy_if(shape.dimensions(), std::back_inserter(new_dims), in StripDegenerateDimensions()
461 return ShapeUtil::MakeShape(shape.element_type(), new_dims); in StripDegenerateDimensions()
|
| /external/eigen/unsupported/Eigen/CXX11/src/Tensor/ |
| D | README.md | 1122 ### <Operation> sum(const Dimensions& new_dims)
1128 ### <Operation> mean(const Dimensions& new_dims)
1134 ### <Operation> maximum(const Dimensions& new_dims)
1140 ### <Operation> minimum(const Dimensions& new_dims)
1146 ### <Operation> prod(const Dimensions& new_dims)
1152 ### <Operation> all(const Dimensions& new_dims)
1158 ### <Operation> any(const Dimensions& new_dims)
1165 ### <Operation> reduce(const Dimensions& new_dims, const Reducer& reducer)
1254 ### <Operation> reshape(const Dimensions& new_dims)
1257 new dimensions. The argument new_dims is an array of Index values. The
[all …]
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_nodes.cc | 2219 nvinfer1::Dims new_dims; in ConvertExpandDims() local
2220 TF_RETURN_IF_ERROR(TensorShapeArrayToTrtDims(input_dims, &new_dims, in ConvertExpandDims()
2224 input_tensor, new_dims, /*validation_only=*/false, &output_tensor)); in ConvertExpandDims()
2280 nvinfer1::Dims new_dims; in ConvertSqueeze() local
2281 TF_RETURN_IF_ERROR(TensorShapeArrayToTrtDims(input_dims, &new_dims, in ConvertSqueeze()
2285 input_tensor, new_dims, /*validation_only=*/false, &output_tensor)); in ConvertSqueeze()
2479 nvinfer1::Dims new_dims; in ConvertStridedSliceHelper() local
2480 TF_RETURN_IF_ERROR(TensorShapeArrayToTrtDims(input_dims, &new_dims, in ConvertStridedSliceHelper()
2484 TRT_TensorOrWeights(tensor), new_dims, /*validation_only=*/false, in ConvertStridedSliceHelper()
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | nn_ops.py | 4031 new_dims = []
4034 new_dims.append(dim.value)
4036 new_dims.append(noise_shape_.dims[i].value)
4037 return tensor_shape.TensorShape(new_dims)
|
| /external/tensorflow/tensorflow/c/ |
| D | c_api.h | 304 const int64_t* new_dims,
|
| D | c_api.cc | 297 TF_Tensor* to, const int64_t* new_dims, in TF_TensorBitcastFrom() argument
313 if (ShapeNumElements(new_dims, num_new_dims) * out_size != in TF_TensorBitcastFrom()
322 p.add_dim()->set_size(new_dims[i]); in TF_TensorBitcastFrom()
|
| D | c_api_test.cc | 1485 int64_t new_dims[] = {3, 2}; in TEST() local
1487 TF_TensorBitcastFrom(a, TF_UINT64, b, new_dims, 2, status); in TEST()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | constant_folding.cc | 1535 PartialTensorShape new_dims; in IsSimplifiableReshape() local
1542 TF_CHECK_OK(TensorShapeUtils::MakeShape(shp, &new_dims)); in IsSimplifiableReshape()
1549 TF_CHECK_OK(TensorShapeUtils::MakeShape(shp, &new_dims)); in IsSimplifiableReshape()
1552 return shape.IsCompatibleWith(new_dims); in IsSimplifiableReshape()
|