| /aosp_15_r20/external/eigen/unsupported/test/ |
| H A D | cxx11_tensor_broadcasting.cpp | 97 array<Index, 3> new_dims; in test_vectorized_broadcasting() local
154 array<Index, 3> new_dims; in test_static_broadcasting() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ops/ |
| H A D | einsum.cc | 376 nvinfer1::Dims* new_dims) { in GetEinsumNewStaticShape()
414 nvinfer1::Dims new_dims; in ConditionEinsumWeights() local
441 nvinfer1::Dims new_dims; in ConditionEinsumTensor() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| H A D | cudnn_vectorize_convolutions.cc | 95 DimensionVector new_dims(shape.dimensions().begin(), in SplitAtDim() local
109 DimensionVector new_dims(shape.dimensions().begin(), in SplitShapeAtDim() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/c/ |
| H A D | tf_tensor.cc | 176 TF_Tensor* to, const int64_t* new_dims, in TF_TensorBitcastFrom()
247 const int64_t* new_dims, int num_new_dims) { in BitcastFrom()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| H A D | tensor_list_ops.cc | 503 std::vector<int64_t> new_dims = {num_elements * tensor_lengths}; in Compile() local
559 std::vector<int64_t> new_dims = {element_dims[0] / length, length}; in Compile() local
|
| H A D | diag_op.cc | 106 std::vector<int64_t> new_dims(dims.size() * 2); in Compile() local
|
| H A D | pooling_ops.cc | 223 absl::InlinedVector<int64, 5> new_dims(result_shape->dimensions().begin(), in Compile() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/ |
| H A D | broadcast_canonicalizer.cc | 40 std::vector<int64_t> new_dims(hlo->dimensions().begin(), in Run() local
|
| H A D | bitcast_decomposer.cc | 150 std::vector<int64_t> new_dims = in Run() local
|
| H A D | scatter_simplifier.cc | 68 std::vector<int64_t> new_dims; in FlattenAndTransposeUpdates() local
|
| H A D | while_loop_concat_code_motion.cc | 832 std::vector<int64_t> new_dims; in RewriteLoopWithConcatGroups() local
|
| H A D | indexed_array_analysis.cc | 464 DimensionVector new_dims; in StripDegenerateDimensions() local
|
| H A D | space_to_batch_converter.cc | 361 new_dims(ctrl_.count_of_dimensions_to_convert); in GetSpatialDimsToSplit() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | transpose_util_test.cc | 30 internal::TransposeDimsVec new_dims; in TestDimensionReduction() local
|
| H A D | transpose_functor.h | 82 TransposeDimsVec* new_dims) { in ReduceTransposeDimensions()
|
| H A D | transpose_functor_gpu.cu.cc | 102 TransposeDimsVec new_dims; in run() local
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/functorch/ |
| H A D | BatchRulesViews.cpp | 142 VmapDimVector new_dims; in flip_batch_rule() local
286 VmapDimVector new_dims; in roll_batch_rule() local
|
| H A D | BatchRulesReduceOps.cpp | 166 VmapDimVector new_dims; in boxed_reduction_batch_rule() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/experimental/acceleration/mini_benchmark/ |
| H A D | decode_jpeg.cc | 93 TfLiteIntArray* new_dims = TfLiteIntArrayCreate(4); in Prepare() local
|
| H A D | call.cc | 71 std::vector<int> new_dims; in ValidateAndResizeInputsIfNeeded() local
|
| /aosp_15_r20/external/pytorch/torch/csrc/lazy/core/ |
| H A D | helpers.cpp | 15 std::vector<int64_t> new_dims; in DropDimensions() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/ |
| H A D | xla_op_kernel.cc | 186 int index, absl::Span<const int64_t> new_dims, in ConstantInputReshaped()
313 int index, absl::Span<const int64_t> new_dims, in ResolveInputDynamismReshaped()
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/toco/graph_transformations/ |
| H A D | resolve_constant_unary.cc | 87 std::vector<int> new_dims; in ReduceGeneric() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
| H A D | dot_handler.cc | 1128 std::vector<int64_t> new_dims; in PartitionBaseCase() local
1287 std::vector<int64_t> new_dims(new_dot_shape.dimensions().begin(), in PartitionBaseCase() local
1310 std::vector<int64_t> new_dims(dot_rhs->shape().dimensions().begin(), in PartitionBaseCase() local
1318 std::vector<int64_t> new_dims(dot_lhs->shape().dimensions().begin(), in PartitionBaseCase() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| H A D | matrix.cc | 559 std::vector<int64_t> new_dims; in Einsum() local
|