Home
last modified time | relevance | path

Searched defs:shape_type (Results 1 – 25 of 27) sorted by relevance

12

/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/hexagon/builders/tests/
H A Dreshape_test.cc34 for (ShapeSpecificationType shape_type : shape_types) { in TYPED_TEST() local
49 for (ShapeSpecificationType shape_type : shape_types) { in TYPED_TEST() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/transforms/
H A Dprepare_tf.cc389 auto shape_type = RankedTensorType::get({4}, rewriter.getIntegerType(32)); in legalizeFilter() local
460 auto shape_type = in RewriteNewAxisMask() local
778 auto shape_type = tf_broadcast_to_op.shape().getType().cast<ShapedType>(); in matchAndRewrite() local
H A Dlower_static_tensor_list.cc604 Type shape_type = RankedTensorType::get({result_rank}, shape_dtype); in matchAndRewrite() local
933 RankedTensorType shape_type = in matchAndRewrite() local
H A Dlegalize_tf.cc860 auto shape_type = in matchAndRewrite() local
/aosp_15_r20/external/eigen/unsupported/Eigen/CXX11/src/Tensor/
H A DTensorBlock.h76 TensorBlockShapeType shape_type; // target block shape member
402 const TensorBlockShapeType shape_type = m_requirements.shape_type; in InitializeBlockDimensions() local
/aosp_15_r20/external/sdv/vsomeip/third_party/boost/endian/example/
Dendian_example.cpp33 little_int32_buf_at shape_type; member
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/
H A Dreshape_test_common.h50 ShapeSpecificationType shape_type) { in ReshapeOpModel()
/aosp_15_r20/external/sdv/vsomeip/third_party/boost/asio/test/execution/
Dexecutor.cpp129 typedef double shape_type; typedef
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/quantization/tensorflow/passes/
H A Dutils.h63 auto shape_type = in CreateConstValue() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/
H A Dfused_ir_emitter.cc111 llvm::Type* shape_type = llvm_ir::ShapeToIrType(constant.shape(), module); in HandleConstant() local
/aosp_15_r20/external/tensorflow/tensorflow/core/transforms/shape_inference/
H A Dpass.cc43 auto shape_type = type.dyn_cast<ShapedType>(); in CanBeRefined() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/
H A Dflatbuffer_import.cc835 auto shape_type = RankedTensorType::get( in ConvertOp() local
902 auto shape_type = RankedTensorType::get( in ConvertOp() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tfrt/transforms/
H A Dcorert_converter.cc304 if (auto shape_type = type_attr.getValue().dyn_cast<mlir::TensorType>()) { in ConvertAttribute() local
/aosp_15_r20/external/sdv/vsomeip/third_party/boost/asio/include/boost/asio/
Dthread_pool.hpp173 typedef std::size_t shape_type; typedef in boost::asio::thread_pool::basic_executor_type
/aosp_15_r20/external/pytorch/torch/csrc/jit/passes/onnx/
H A Dunpack_quantized_weights.cpp557 TensorTypePtr shape_type = g_input->type()->cast<TensorType>(); in UnpackQuantizedTensorInputs() local
H A Dshape_type_inference.cpp693 if (TensorTypePtr shape_type = input->type()->cast<TensorType>()) { in SetShapeValueFromListConstructNode() local
1007 if (TensorTypePtr shape_type = n->input(1)->type()->cast<TensorType>()) { in ProcessReshapeNode() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
H A Dtensor_list_ops_decomposition.cc788 auto shape_type = RankedTensorType::get({2}, builder.getIntegerType(32)); in HandleTensorListScatterIntoExistingListOp() local
H A Dlower_tf.cc518 auto shape_type = RankedTensorType::get({2}, rewriter.getIntegerType(32)); in matchAndRewrite() local
H A Dshape_inference.cc416 auto shape_type = type.dyn_cast<ShapedType>(); in CanBeRefined() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tosa/transforms/
H A Dlegalize_tf.cc722 RankedTensorType shape_type = RankedTensorType::get( in matchAndRewrite() local
H A Dlegalize_tfl.cc1711 RankedTensorType shape_type = RankedTensorType::get( in matchAndRewrite() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/toco/
H A Dexport_tensorflow.cc1968 const tensorflow::DataType shape_type = in ConvertRandomUniformOperator() local
/aosp_15_r20/external/tensorflow/tensorflow/core/transforms/constant_folding/
H A Dpass.cc1779 auto shape_type = op->getOperand(0).getType().cast<ShapedType>(); in matchAndRewrite() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tests/
H A Drandomized_tests.cc3714 auto shape_type = Choose<DataType>({DT_INT32, DT_INT64}); in TEST_F() local
3727 auto shape_type = Choose<DataType>({DT_INT32, DT_INT64}); in TEST_F() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/
H A Dtf_ops_n_z.cc2154 auto shape_type = op.shape().getType().dyn_cast<RankedTensorType>(); in verify() local

12