| /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/core/kernels/hexagon/ |
| D | hexagon_control_wrapper.cc | 108 const RemoteFusedGraphExecuteInfo::TensorShapeTypeProto& shape_type = in Init() local
119 const RemoteFusedGraphExecuteInfo::TensorShapeTypeProto& shape_type = in Init() local
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/transforms/ |
| D | prepare_tf.cc | 492 auto shape_type = RankedTensorType::get({4}, rewriter.getIntegerType(32)); in legalizeFilter() local
556 auto shape_type = in RewriteNewAxisMask() local
857 auto shape_type = tf_broadcast_to_op.shape().getType().cast<ShapedType>(); in matchAndRewrite() local
|
| D | lower_static_tensor_list.cc | 460 Type shape_type = RankedTensorType::get({result_rank}, shape_dtype); in matchAndRewrite() local
770 RankedTensorType shape_type = in matchAndRewrite() local
|
| D | legalize_tf.cc | 688 auto shape_type = in matchAndRewrite() local
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/ |
| D | flatbuffer_import.cc | 676 auto shape_type = RankedTensorType::get( in ConvertOp() local
742 auto shape_type = RankedTensorType::get( in ConvertOp() local
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | shape_inference.cc | 81 auto shape_type = type.dyn_cast<ShapedType>(); in CanBeRefined() local
102 auto shape_type = potential_refined_type.dyn_cast<ShapedType>(); in CanRefineTypeWith() local
|
| D | tensor_list_ops_decomposition.cc | 798 auto shape_type = RankedTensorType::get({2}, builder.getIntegerType(32)); in HandleTensorListScatterIntoExistingListOp() local
|
| D | lower_tf.cc | 485 auto shape_type = RankedTensorType::get({2}, rewriter.getIntegerType(32)); in matchAndRewrite() local
|
| /external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/ |
| D | legalize_to_linalg.cc | 501 auto shape_type = shape.getType().cast<RankedTensorType>(); in matchAndRewrite() local
|
| /external/tensorflow/tensorflow/compiler/mlir/tosa/transforms/ |
| D | legalize_tf.cc | 717 RankedTensorType shape_type = RankedTensorType::get( in matchAndRewrite() local
|
| D | legalize_tfl.cc | 1635 RankedTensorType shape_type = RankedTensorType::get( in matchAndRewrite() local
|
| /external/tensorflow/tensorflow/lite/toco/ |
| D | export_tensorflow.cc | 1963 const tensorflow::DataType shape_type = in ConvertRandomUniformOperator() local
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/ir/ |
| D | tfl_ops.cc | 1059 auto shape_type = shape.getType().cast<ShapedType>(); in Verify() local
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_ops_n_z.cc | 2013 auto shape_type = op.shape().getType().dyn_cast<RankedTensorType>(); in Verify() local
|