| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2tensorrt/utils/ |
| H A D | trt_shape_optimization_profiles.cc | 36 std::vector<nvinfer1::Dims> GetDimVec(std::vector<TensorShapeType> shape_vec) { in GetDimVec()
73 for (auto& shape_vec : collected_shapes) { in ImplicitBatchModeCompatibleStrategy() local
129 for (auto& shape_vec : collected_shapes) { in OptimalStrategy() local
271 auto shape_vec = input_shapes_[i]; in InitProfiles() local
329 auto shape_vec = shapes; in InitCalibProfile() local
|
| H A D | trt_shape_optimization_profiles_test.cc | 113 std::vector<TensorShape> shape_vec = DimVecToShapeVec(dimvec); in CheckProfile() local
180 std::vector<TensorShape> shape_vec = DimVecToShapeVec(dim_vec); in TEST_P() local
206 std::vector<TensorShape> shape_vec = DimVecToShapeVec(dim_vec, true); in TEST_P() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | avgpooling_op.cc | 305 auto shape_vec = tensor_in_shape.vec<int32>(); in Compute() local
464 auto shape_vec = tensor_in_shape.vec<int32>(); in Compute() local
550 auto shape_vec = tensor_in_shape.vec<int32>(); in Compute() local
|
| H A D | sparse_utils.cc | 203 const auto shape_vec = shape.flat<Tindices>(); in ValidateSparseTensorIndicesUnordered() local
227 const auto shape_vec = shape.flat<Tindices>(); in ValidateSparseTensorIndicesOrdered() local
|
| H A D | sparse_dense_binary_op_shared.cc | 101 const auto shape_vec = shape_t->vec<int64_t>(); in Compute() local
|
| H A D | sparse_reduce_op.cc | 176 const auto shape_vec = shape_t->vec<int64_t>(); in Compute() local
|
| H A D | pooling_ops_3d.cc | 524 auto shape_vec = tensor_in_shape.vec<int32>(); in Compute() local
|
| H A D | sparse_utils_test.cc | 338 auto shape_vec = output_shape.vec<int64_t>(); in GenerateRandomSparseTensor() local
|
| H A D | sparse_cross_op.cc | 732 auto shape_vec = (*shape_out)->vec<int64_t>(); in CreateOutputTensors() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/profiler/internal/ |
| H A D | tfprof_node.cc | 258 std::vector<int64_t> shape_vec; in ShapeProtoToVec() local
270 TensorShapeProto VecToShapeProto(const std::vector<int64_t>& shape_vec) { in VecToShapeProto()
|
| H A D | tfprof_show.cc | 227 std::vector<string> shape_vec; in FormatNode() local
|
| H A D | tfprof_node.h | 708 std::vector<int64_t>& shape_vec = input_shapes[inp.first]; in input_shapes() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/image/ |
| H A D | random_crop_op.cc | 47 auto shape_vec = shape_t.vec<int64_t>(); in Compute() local
|
| H A D | image_ops.cc | 77 auto shape_vec = shape_t.vec<int32>(); in DoImageProjectiveTransformOp() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/tools/versioning/ |
| H A D | op_signature.cc | 65 const flatbuffers::Vector<int32_t>* shape_vec = in GetOpSignatureTensorSpecs() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2tensorrt/kernels/ |
| H A D | trt_engine_resource_ops_test.cc | 170 std::vector<TensorShape> shape_vec(n_input); in CreateTRTEngine() local
|
| H A D | trt_engine_op_test.cc | 116 std::vector<PartialTensorShape> shape_vec{shape, {}}; local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/mkl/ |
| H A D | mkl_avgpooling_op.cc | 204 auto shape_vec = orig_input_tensor.vec<int32>(); in Compute() local
|
| /aosp_15_r20/external/executorch/extension/android/jni/ |
| H A D | jni_layer.cpp | 178 std::vector<exec_aten::SizesType> shape_vec; in JEValueToTensorImpl() local
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/tensorexpr/ |
| H A D | graph_opt.cpp | 247 std::vector<at::ShapeSymbol> shape_vec = *tt->symbolic_sizes().sizes(); in makeShapesSymbolic() local
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/passes/ |
| H A D | symbolic_shape_runtime_fusion.cpp | 195 std::vector<at::ShapeSymbol> shape_vec = *tt.symbolic_sizes().sizes(); in TryGeneralizeInputDimensionsToSymbolicShapes() local
|
| H A D | symbolic_shape_analysis.cpp | 927 std::vector<at::ShapeSymbol> shape_vec = *tt->symbolic_sizes().sizes(); in updateGraphWithSymbolicShapeEqualities() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tosa/transforms/ |
| H A D | legalize_tf.cc | 1011 SmallVector<int64_t> shape_vec; in matchAndRewrite() local
|
| H A D | legalize_common.cc | 2692 SmallVector<int64_t> shape_vec(input_shape.begin(), input_shape.end()); in convertReduceOpCommon() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/framework/ |
| H A D | common_shape_fns.cc | 2514 auto shape_vec = shape_tensor->flat<int64_t>(); in SparseReduceShapeFn() local
|