| /external/tensorflow/tensorflow/lite/kernels/ |
| D | reshape_test.cc | 47 for (ShapeSpecificationType shape_type : in TYPED_TEST()
49 if (shape_type == ShapeSpecificationType::kAsTensor) { in TYPED_TEST()
50 ReshapeOpModel<TypeParam> m({1, 2, 4, 1}, {2}, {2, 1}, shape_type); in TYPED_TEST()
57 ReshapeOpModel<TypeParam>({1, 2, 4, 1}, {2}, {2, 1}, shape_type), in TYPED_TEST()
65 for (ShapeSpecificationType shape_type : in TYPED_TEST()
70 {1, 1, 1, 1, 1, 1, 1, 1, 2}, shape_type), in TYPED_TEST()
77 for (ShapeSpecificationType shape_type : in TYPED_TEST()
79 if (shape_type != ShapeSpecificationType::kAsTensor) { in TYPED_TEST()
82 shape_type), in TYPED_TEST()
87 shape_type); in TYPED_TEST()
[all …]
|
| D | reshape_test_common.h | 50 ShapeSpecificationType shape_type) { in ReshapeOpModel() argument
51 switch (shape_type) { in ReshapeOpModel()
|
| /external/tensorflow/tensorflow/lite/delegates/hexagon/builders/tests/ |
| D | reshape_test.cc | 34 for (ShapeSpecificationType shape_type : shape_types) { in TYPED_TEST() local
36 {1, 2, 4, 1}, {3}, {2, 2, 2}, shape_type); in TYPED_TEST()
49 for (ShapeSpecificationType shape_type : shape_types) { in TYPED_TEST() local
51 {1, 2, 4, 1}, {3}, {2, 1, -1}, shape_type); in TYPED_TEST()
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/ |
| D | TensorListElementShape.pbtxt | 9 type_attr: "shape_type"
12 name: "shape_type"
|
| D | EmptyTensorList.pbtxt | 5 type_attr: "shape_type"
20 name: "shape_type"
|
| D | TensorListFromTensor.pbtxt | 9 type_attr: "shape_type"
20 name: "shape_type"
|
| D | TensorListReserve.pbtxt | 5 type_attr: "shape_type"
20 name: "shape_type"
|
| D | TensorListScatter.pbtxt | 13 type_attr: "shape_type"
24 name: "shape_type"
|
| D | TensorListSplit.pbtxt | 9 type_attr: "shape_type"
24 name: "shape_type"
|
| D | TensorListConcatV2.pbtxt | 9 type_attr: "shape_type"
28 name: "shape_type"
|
| D | TensorListScatterV2.pbtxt | 13 type_attr: "shape_type"
28 name: "shape_type"
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/ |
| D | TensorListElementShape.pbtxt | 9 type_attr: "shape_type"
12 name: "shape_type"
|
| D | TensorListFromTensor.pbtxt | 9 type_attr: "shape_type"
20 name: "shape_type"
|
| D | EmptyTensorList.pbtxt | 5 type_attr: "shape_type"
20 name: "shape_type"
|
| D | TensorListReserve.pbtxt | 5 type_attr: "shape_type"
20 name: "shape_type"
|
| D | TensorListSplit.pbtxt | 9 type_attr: "shape_type"
24 name: "shape_type"
|
| D | TensorListScatter.pbtxt | 13 type_attr: "shape_type"
24 name: "shape_type"
|
| D | TensorListConcatV2.pbtxt | 9 type_attr: "shape_type"
28 name: "shape_type"
|
| D | TensorListScatterV2.pbtxt | 13 type_attr: "shape_type"
28 name: "shape_type"
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | list_ops.py | 228 op.outputs[0], shape_type=dtypes.int32))
246 op.inputs[0], shape_type=dtypes.int32),
278 op.outputs[0], shape_type=dtypes.int32))
295 shape_type=dtypes.int32),
331 input_list, shape_type=dtypes.int32)
|
| /external/tensorflow/tensorflow/core/kernels/hexagon/ |
| D | hexagon_control_wrapper.cc | 108 const RemoteFusedGraphExecuteInfo::TensorShapeTypeProto& shape_type = in Init() local
110 int64 buf_size = DataTypeSize(shape_type.dtype()); in Init()
111 for (const TensorShapeProto::Dim& dim : shape_type.shape().dim()) { in Init()
119 const RemoteFusedGraphExecuteInfo::TensorShapeTypeProto& shape_type = in Init() local
122 int64 buf_size = DataTypeSize(shape_type.dtype()); in Init()
123 for (const TensorShapeProto::Dim& dim : shape_type.shape().dim()) { in Init()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/graphdef2mlir/ |
| D | tensor-list.pbtxt | 142 key: "shape_type"
162 key: "shape_type"
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/transforms/ |
| D | prepare_tf.cc | 492 auto shape_type = RankedTensorType::get({4}, rewriter.getIntegerType(32)); in legalizeFilter() local
498 auto shape_attr = DenseElementsAttr::get(shape_type, result_shape_data); in legalizeFilter()
499 auto shape = rewriter.create<TF::ConstOp>(loc, shape_type, shape_attr); in legalizeFilter()
556 auto shape_type = in RewriteNewAxisMask() local
564 auto shape_attr = DenseElementsAttr::get(shape_type, result_shape_data); in RewriteNewAxisMask()
565 auto shape = rewriter.create<ConstantOp>(loc, shape_type, shape_attr); in RewriteNewAxisMask()
857 auto shape_type = tf_broadcast_to_op.shape().getType().cast<ShapedType>(); in matchAndRewrite() local
863 (shape_type.hasStaticShape() && shape_type.getRank() == 1 && in matchAndRewrite()
864 shape_type.getDimSize(0) <= 4))) in matchAndRewrite()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | shape_inference.cc | 81 auto shape_type = type.dyn_cast<ShapedType>(); in CanBeRefined() local
82 if (!shape_type) return false; in CanBeRefined()
90 shape_type.getElementType().dyn_cast<TF::TensorFlowTypeWithSubtype>(); in CanBeRefined()
93 return !shape_type.hasStaticShape(); in CanBeRefined()
102 auto shape_type = potential_refined_type.dyn_cast<ShapedType>(); in CanRefineTypeWith() local
103 if (!shape_type) return false; in CanRefineTypeWith()
104 if (shape_type.hasRank()) return true; in CanRefineTypeWith()
107 shape_type.getElementType().dyn_cast<TF::TensorFlowTypeWithSubtype>(); in CanRefineTypeWith()
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | tensor_list_ops_test.py | 43 e32 = list_ops.tensor_list_element_shape(l, shape_type=dtypes.int32)
44 e64 = list_ops.tensor_list_element_shape(l, shape_type=dtypes.int64)
|