| /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/core/ops/compat/ops_history_v2/ |
| D | TensorListFromTensor.pbtxt | 9 type_attr: "shape_type"
20 name: "shape_type"
38 type_attr: "shape_type"
56 name: "shape_type"
74 type_attr: "shape_type"
95 name: "shape_type"
|
| D | EmptyTensorList.pbtxt | 5 type_attr: "shape_type"
20 name: "shape_type"
34 type_attr: "shape_type"
56 name: "shape_type"
70 type_attr: "shape_type"
95 name: "shape_type"
|
| D | TensorListReserve.pbtxt | 5 type_attr: "shape_type"
20 name: "shape_type"
34 type_attr: "shape_type"
56 name: "shape_type"
70 type_attr: "shape_type"
95 name: "shape_type"
|
| D | TensorListScatter.pbtxt | 13 type_attr: "shape_type"
24 name: "shape_type"
46 type_attr: "shape_type"
64 name: "shape_type"
86 type_attr: "shape_type"
107 name: "shape_type"
|
| D | TensorListSplit.pbtxt | 9 type_attr: "shape_type"
24 name: "shape_type"
42 type_attr: "shape_type"
64 name: "shape_type"
82 type_attr: "shape_type"
107 name: "shape_type"
|
| D | TensorListScatterV2.pbtxt | 13 type_attr: "shape_type"
28 name: "shape_type"
50 type_attr: "shape_type"
72 name: "shape_type"
94 type_attr: "shape_type"
119 name: "shape_type"
|
| D | TensorListElementShape.pbtxt | 9 type_attr: "shape_type"
12 name: "shape_type"
|
| D | TensorListConcatV2.pbtxt | 9 type_attr: "shape_type"
28 name: "shape_type"
|
| /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 | TensorListReserve.pbtxt | 5 type_attr: "shape_type"
20 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 | TensorListScatter.pbtxt | 13 type_attr: "shape_type"
24 name: "shape_type"
|
| D | TensorListSplit.pbtxt | 9 type_attr: "shape_type"
24 name: "shape_type"
|
| D | TensorListScatterV2.pbtxt | 13 type_attr: "shape_type"
28 name: "shape_type"
|
| D | TensorListConcatV2.pbtxt | 9 type_attr: "shape_type"
28 name: "shape_type"
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | list_ops.py | 225 op.outputs[0], shape_type=dtypes.int32))
243 op.inputs[0], shape_type=dtypes.int32),
275 op.outputs[0], shape_type=dtypes.int32))
292 shape_type=dtypes.int32),
328 input_list, shape_type=dtypes.int32)
|
| /external/tensorflow/tensorflow/core/transforms/shape_inference/ |
| D | pass.cc | 43 auto shape_type = type.dyn_cast<ShapedType>(); in CanBeRefined() local
44 if (!shape_type) return false; in CanBeRefined()
51 auto type_with_subtype = shape_type.getElementType() in CanBeRefined()
55 return !shape_type.hasStaticShape(); in CanBeRefined()
|
| /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
114 global, shape_type->getPointerTo()); in HandleConstant()
116 IrArray array(global_with_shape, shape_type, constant.shape()); in HandleConstant()
|
| /external/tensorflow/tensorflow/core/ir/importexport/tests/roundtrip/ |
| D | tensor-list.pbtxt | 140 key: "shape_type"
160 key: "shape_type"
|
| /external/eigen/unsupported/Eigen/CXX11/src/Tensor/ |
| D | TensorBlock.h | 76 TensorBlockShapeType shape_type; // target block shape member
87 : shape_type(shape_type_), size(size_), cost_per_coeff(cost_) in TensorBlockResourceRequirements()
93 TensorBlockShapeType shape_type, size_t size_in_bytes, in withShapeAndSize()
96 return {shape_type, size, cost}; in withShapeAndSize()
101 TensorBlockShapeType shape_type, size_t size_in_bytes) { in withShapeAndSize()
116 return withShapeAndSize<Scalar>(shape_type, size_in_bytes, in withShapeAndSize()
140 return {merge(lhs.shape_type, rhs.shape_type), // shape_type in merge()
402 const TensorBlockShapeType shape_type = m_requirements.shape_type; in InitializeBlockDimensions() local
436 if (shape_type == TensorBlockShapeType::kSkewedInnerDims) { in InitializeBlockDimensions()
449 } else if (shape_type == TensorBlockShapeType::kUniformAllDims) { in InitializeBlockDimensions()
|
| /external/tensorflow/tensorflow/compiler/mlir/quantization/tensorflow/passes/ |
| D | utils.h | 63 auto shape_type = in CreateConstValue() local
66 DenseIntElementsAttr attr = DenseIntElementsAttr::get(shape_type, values); in CreateConstValue()
|