| /external/tensorflow/tensorflow/lite/delegates/hexagon/builders/tests/ |
| D | reshape_test.cc | 34 for (ShapeSpecificationType shape_type : shape_types) { in TYPED_TEST() local
49 for (ShapeSpecificationType shape_type : shape_types) { in TYPED_TEST() local
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | reshape_test_common.h | 50 ShapeSpecificationType shape_type) { in ReshapeOpModel()
|
| /external/tensorflow/tensorflow/compiler/mlir/quantization/tensorflow/passes/ |
| D | utils.h | 63 auto shape_type = in CreateConstValue() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/ |
| D | fused_ir_emitter.cc | 111 llvm::Type* shape_type = llvm_ir::ShapeToIrType(constant.shape(), module); in HandleConstant() local
|
| /external/eigen/unsupported/Eigen/CXX11/src/Tensor/ |
| D | TensorBlock.h | 76 TensorBlockShapeType shape_type; // target block shape member
402 const TensorBlockShapeType shape_type = m_requirements.shape_type; in InitializeBlockDimensions() local
|
| /external/tensorflow/tensorflow/core/transforms/shape_inference/ |
| D | pass.cc | 43 auto shape_type = type.dyn_cast<ShapedType>(); in CanBeRefined() local
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/transforms/ |
| D | prepare_tf.cc | 389 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
|
| D | lower_static_tensor_list.cc | 604 Type shape_type = RankedTensorType::get({result_rank}, shape_dtype); in matchAndRewrite() local
933 RankedTensorType shape_type = in matchAndRewrite() local
|
| D | legalize_tf.cc | 860 auto shape_type = in matchAndRewrite() local
|
| /external/tensorflow/tensorflow/compiler/mlir/tfrt/transforms/ |
| D | corert_converter.cc | 304 if (auto shape_type = type_attr.getValue().dyn_cast<mlir::TensorType>()) { in ConvertAttribute() local
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/ |
| D | flatbuffer_import.cc | 835 auto shape_type = RankedTensorType::get( in ConvertOp() local
902 auto shape_type = RankedTensorType::get( in ConvertOp() local
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | tensor_list_ops_decomposition.cc | 788 auto shape_type = RankedTensorType::get({2}, builder.getIntegerType(32)); in HandleTensorListScatterIntoExistingListOp() local
|
| D | lower_tf.cc | 518 auto shape_type = RankedTensorType::get({2}, rewriter.getIntegerType(32)); in matchAndRewrite() local
|
| D | shape_inference.cc | 416 auto shape_type = type.dyn_cast<ShapedType>(); in CanBeRefined() local
|
| /external/tensorflow/tensorflow/compiler/mlir/tosa/transforms/ |
| D | legalize_tf.cc | 722 RankedTensorType shape_type = RankedTensorType::get( in matchAndRewrite() local
|
| D | legalize_tfl.cc | 1711 RankedTensorType shape_type = RankedTensorType::get( in matchAndRewrite() local
|
| /external/tensorflow/tensorflow/lite/toco/ |
| D | export_tensorflow.cc | 1968 const tensorflow::DataType shape_type = in ConvertRandomUniformOperator() local
|
| /external/tensorflow/tensorflow/core/transforms/constant_folding/ |
| D | pass.cc | 1779 auto shape_type = op->getOperand(0).getType().cast<ShapedType>(); in matchAndRewrite() local
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_ops_n_z.cc | 2154 auto shape_type = op.shape().getType().dyn_cast<RankedTensorType>(); in verify() local
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | randomized_tests.cc | 3714 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
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/ir/ |
| D | tfl_ops.cc | 1441 auto shape_type = shape.getType().cast<ShapedType>(); in verify() local
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | legalize_tf.cc | 3182 auto shape_type = shape::ShapeType::get(rewriter->getContext()); in BroadcastBatchMatMulV2Operands() local
|