Home
last modified time | relevance | path

Searched refs:const_shape (Results 1 – 11 of 11) sorted by relevance

/external/llvm-project/mlir/test/Dialect/Shape/
Dcanonicalize.mlir5 // CHECK: shape.const_shape [2, 3, 4] : tensor<?xindex>
15 // CHECK: shape.const_shape [2, 3] : !shape.shape
16 // CHECK: shape.const_shape [4, 5] : !shape.shape
18 %0 = shape.const_shape [2, 3, 4, 5] : !shape.shape
29 // CHECK: shape.const_shape [2, 3, 4] : !shape.shape
30 // CHECK: shape.const_shape [5] : !shape.shape
32 %0 = shape.const_shape [2, 3, 4, 5] : !shape.shape
44 %0 = shape.const_shape [2, 3, 4, 5] : !shape.shape
54 // CHECK: shape.const_shape [7, 2] : !shape.shape
55 %0 = shape.const_shape [1, 2] : !shape.shape
[all …]
Dops.mlir36 func @const_shape() {
37 %0 = shape.const_shape [1, 2, 3] : !shape.shape
38 %1 = shape.const_shape [4, 5, 6] : tensor<?xindex>
43 %0 = shape.const_shape [1, 57, 92] : !shape.shape
50 %0 = shape.const_shape [10, 1, 57, 92] : !shape.shape
51 %1 = shape.const_shape [4, 57, 92] : !shape.shape
58 %0 = shape.const_shape [10, 1, 57, 92] : tensor<?xindex>
59 %1 = shape.const_shape [4, 57, 92] : tensor<?xindex>
65 %0 = shape.const_shape [4, 57, 92] : !shape.shape
66 %1 = shape.const_shape [4, 57, 92] : !shape.shape
[all …]
/external/llvm-project/mlir/test/Conversion/ShapeToStandard/
Dshape-to-standard.mlir90 // Lower `const_shape` to `tensor_from_elements`.
91 // CHECK-LABEL: @const_shape
93 func @const_shape() -> tensor<?xindex> {
100 %shape = shape.const_shape [1, 2, 3] : tensor<?xindex>
106 // Lower `const_shape` in the case of rank 0.
113 %shape = shape.const_shape [] : tensor<?xindex>
/external/tensorflow/tensorflow/compiler/mlir/hlo/tests/
Dhlo-transform-unranked.mlir209 // CHECK-NEXT: %[[CONST_SHAPE_1:.*]] = shape.const_shape [1]
224 // CHECK-NEXT: %[[CONST_SHAPE_2:.*]] = shape.const_shape [1, 1]
239 // CHECK-NEXT: %[[CONST_SHAPE_3:.*]] = shape.const_shape [1, 1, 1]
254 // CHECK-NEXT: %[[CONST_SHAPE_4:.*]] = shape.const_shape [1, 1, 1, 1]
269 // CHECK-NEXT: %[[CONST_SHAPE_5:.*]] = shape.const_shape [1, 1, 1, 1, 1]
284 // CHECK-NEXT: %[[CONST_SHAPE_6:.*]] = shape.const_shape [1, 1, 1, 1, 1, …
/external/tensorflow/tensorflow/python/ops/
Drandom_ops.py544 const_shape = tensor_util.constant_value_as_shape(shape)
546 tensor.set_shape(const_shape.concatenate(postfix_tensor.shape))
/external/llvm-project/mlir/include/mlir/Dialect/Shape/IR/
DShapeOps.td88 def Shape_ConstShapeOp : Shape_Op<"const_shape", [ConstantLike, NoSideEffect]> {
96 %0 = shape.const_shape [] : !shape.shape
97 %1 = shape.const_shape [1, 2, 3] : !shape.shape
98 %2 = shape.const_shape [4, 5, 6] : tensor<?xindex>
744 %0 = shape.const_shape [1,2,3]
745 %1 = shape.const_shape [1, 2, 3]
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/
Dtf_ops_n_z.cc2468 llvm::SmallVector<int64_t, 4> const_shape; in build() local
2470 const_shape.assign( in build()
2474 const_shape.reserve(attr_shape.getNumElements()); in build()
2476 const_shape.push_back(x_type.getDimSize(dim.getSExtValue())); in build()
2479 builder, result, RankedTensorType::get(const_shape, etype), x, perm); in build()
/external/tensorflow/tensorflow/python/framework/
Dtensor_util.py1118 const_shape = constant_value_as_shape(shape)
1119 tensor.set_shape(const_shape)
/external/tensorflow/tensorflow/compiler/mlir/tfr/tests/
Dops.mlir83 // CANON-NEXT: %[[s:.*]] = shape.const_shape [1, 2] : tensor<?xindex>
/external/tensorflow/tensorflow/compiler/mlir/xla/tests/
Dlegalize-tf-binary-elementwise.mlir215 // NOT-CHECK-DAG: %[[RHS_SHAPE:.+]] = shape.const_shape [1]
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dconstant_folding.cc3517 const auto& const_shape = const_props[0].shape(); in MulConvPushDown() local
3519 conv_node->attr().at("data_format").s(), filter_shape, const_shape)) { in MulConvPushDown()