Home
last modified time | relevance | path

Searched refs:operand_segment_sizes (Results 1 – 25 of 31) sorted by relevance

12

/external/llvm-project/mlir/test/Dialect/OpenMP/
Dops.mlir62 …}) {operand_segment_sizes = dense<[0,1,1,1,1,1,1,1]>: vector<8xi32>, default_val = "defshared"} : …
71 …}) {operand_segment_sizes = dense<[1,0,1,1,1,1,1,1]> : vector<8xi32>} : (i1, memref<i32>, memref<i…
77 …}) {operand_segment_sizes = dense<[1,1,1,1,1,1,0,0]> : vector<8xi32>} : (i1, si32, memref<i32>, me…
80 …}) {operand_segment_sizes = dense<[1,1,1,1,1,1,1,1]> : vector<8xi32>, proc_bind_val = "spread"} : …
86 …}) {operand_segment_sizes = dense<[0,0,1,2,1,1,1,1]> : vector<8xi32>} : (memref<i32>, memref<i32>,…
135 …}) {operand_segment_sizes = dense<[1,1,1,1,0,0,0,0,0]> : vector<9xi32>, collapse_val = 2, ordered_…
141 …}) {operand_segment_sizes = dense<[2,2,2,1,0,0,0,0,0]> : vector<9xi32>, collapse_val = 2, ordered_…
148 }) {operand_segment_sizes = dense<[1,1,1,0,0,0,1,1,0]> : vector<9xi32>, schedule_val = "Static"} :
154 …}) {operand_segment_sizes = dense<[1,1,1,1,1,1,1,1,1]> : vector<9xi32>, schedule_val = "Dynamic", …
160 …}) {operand_segment_sizes = dense<[1,1,1,1,0,0,0,0,0]> : vector<9xi32>, nowait, schedule_val = "Au…
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/
Dtf_device_ops_invalid.mlir64 }) {n = 2 : i32, operand_segment_sizes = dense<[0, 0]> : vector<2xi32>} : () -> ()
75 }) {n = 2 : i32, operand_segment_sizes = dense<[0, 0]> : vector<2xi32>} : () -> ()
88 }) {n = 2 : i32, operand_segment_sizes = dense<[0, 0]> : vector<2xi32>} : () -> ()
100 }) {n = 1 : i32, operand_segment_sizes = dense<[0, 0]> : vector<2xi32>} : () -> ()
112 …TPU_REPLICATED_CORE_0 = ["/DEVICE:0", "/DEVICE:1"]}, n = 3 : i32, operand_segment_sizes = dense<[0…
124 …/DEVICE:1"], TPU_REPLICATED_CORE_1 = ["/DEVICE:2"]}, n = 3 : i32, operand_segment_sizes = dense<[0…
135 …}) {n = 3 : i32, operand_segment_sizes = dense<[4, 0]> : vector<2xi32>} : (tensor<*xi32>, tensor<*…
147 …}) {n = 3 : i32, operand_segment_sizes = dense<[3, 2]> : vector<2xi32>} : (tensor<*xi32>, tensor<*…
159 …}) {n = 2 : i32, operand_segment_sizes = dense<[2, 0]> : vector<2xi32>} : (tensor<*xi32>, tensor<*…
171 }) {n = 2 : i32, operand_segment_sizes = dense<[0, 1]> : vector<2xi32>} : (tensor<*xi1>) -> ()
[all …]
Drewrite_tpu_embedding_ops.mlir18 …f._SendTPUEmbeddingGradients"(%arg0, %[[DATA]]) {config = {{.*}}, operand_segment_sizes = dense<[1…
22 … \01*\0C\1A\00j\05\0D\00\00\80?\88\01\01\10\02\18\80\04 \01(\02", operand_segment_sizes = dense<[1…
33 … \01*\0C\1A\00j\05\0D\00\00\80?\88\01\01\10\02\18\80\04 \01(\02", operand_segment_sizes = dense<[1…
Dtf_device_ops.mlir100 tf_device.replicate {n = 2 : i32, operand_segment_sizes = dense<[0, 0]> : vector<2xi32>} {
106 // CHECK-NOT: operand_segment_sizes
Dtpu_update_embedding_enqueue_op_inputs.mlir44 …%2, %3) {_tpu_embedding_layer = "call1", config = "\0A\0B\0C\0D", operand_segment_sizes = dense<[2…
Dtpu_extract_head_tail_outside_compilation.mlir507 …N = 1 : i64, NN = 0 : i64, config = "test_config_send_embedding", operand_segment_sizes = dense<[1…
532 …N = 1 : i64, NN = 0 : i64, config = "test_config_send_embedding", operand_segment_sizes = dense<[1…
Dmark_ops_for_outside_compilation.mlir89 …, %2#1) {_tpu_embedding_layer = "call1", config = "\0A\0B\0C\0D", operand_segment_sizes = dense<[2…
/external/llvm-project/mlir/test/IR/
Dtraits.mlir366 // expected-error @+1 {{requires 1D vector attribute 'operand_segment_sizes'}}
373 // expected-error @+1 {{requires 1D vector attribute 'operand_segment_sizes'}}
374 …"test.attr_sized_operands"(%arg, %arg, %arg, %arg) {operand_segment_sizes = dense<[1, 1, 1, 1]>: t…
380 // expected-error @+1 {{requires 1D vector attribute 'operand_segment_sizes'}}
381 …"test.attr_sized_operands"(%arg, %arg, %arg, %arg) {operand_segment_sizes = dense<[[1, 1], [1, 1]]…
387 // expected-error @+1 {{'operand_segment_sizes' attribute cannot have negative elements}}
388 …"test.attr_sized_operands"(%arg, %arg, %arg, %arg) {operand_segment_sizes = dense<[1, 1, -1, 1]>: …
394 …d count (4) does not match with the total size (3) specified in attribute 'operand_segment_sizes'}}
395 …"test.attr_sized_operands"(%arg, %arg, %arg, %arg) {operand_segment_sizes = dense<[0, 1, 1, 1]>: v…
401 …// expected-error @+1 {{'operand_segment_sizes' attribute for specifying operand segments must hav…
[all …]
/external/llvm-project/mlir/test/mlir-tblgen/
Dop-python-bindings.td26 // CHECK: operand_segment_sizes = array.array('L')
28 // CHECK: operand_segment_sizes.append(len(variadic1))
30 // CHECK: operand_segment_sizes.append(1)
32 // CHECK: operand_segment_sizes.append(0 if variadic2 is None else 1)
33 // CHECK: attributes["operand_segment_sizes"] = _ir.DenseElementsAttr.get(operand_segment_sizes,
43 // CHECK: self.operation.attributes["operand_segment_sizes"], 0)
50 // CHECK: self.operation.attributes["operand_segment_sizes"], 1)
57 // CHECK: self.operation.attributes["operand_segment_sizes"], 2)
Dop-decl.td110 I32ElementsAttr:$operand_segment_sizes
121 // CHECK: ::mlir::DenseIntElementsAttr operand_segment_sizes();
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/
Dtf_device.cc382 (*op.operand_segment_sizes().int_value_begin()).getSExtValue(); in Print()
450 auto operand_segment_sizes = op.operand_segment_sizes(); in Verify() local
452 operand_segment_sizes.getValue<IntegerAttr>({0}).getInt(); in Verify()
454 operand_segment_sizes.getValue<IntegerAttr>({1}).getInt(); in Verify()
547 auto operand_segment_sizes = in BuildReplicateOp() local
550 state->addAttribute(kOperandSegmentSizesAttr, operand_segment_sizes); in BuildReplicateOp()
Dtf_device_ops.td271 I32ElementsAttr:$operand_segment_sizes,
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/graphdef2mlir/
Dparse_example.pbtxt4 # CHECK: operand_segment_sizes = dense<[1, 1, 0, 1, 1]> : vector<5xi32>
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/mlir2graphdef/
Dtf_tpu_embedding_ops.mlir6 …N = 1 : i64, NN = 0 : i64, config = "test_config_send_embedding", operand_segment_sizes = dense<[1…
Dparse_example.mlir56 …tputs_0, %outputs_2) {dense_shapes = [#tf.shape<1>], device = "", operand_segment_sizes = dense<[1…
/external/llvm-project/mlir/test/Dialect/PDL/
Dinvalid.mlir97 operand_segment_sizes = dense<0> : vector<3xi32>
175 operand_segment_sizes = dense<1> : vector<3xi32>
/external/llvm-project/mlir/test/Conversion/OpenMPToLLVM/
Dconvert-to-llvmir.mlir45 …}) {operand_segment_sizes = dense<[2, 2, 2, 0, 0, 0, 0, 0, 0]> : vector<9xi32>} : (index, index, i…
/external/llvm-project/mlir/test/Dialect/GPU/
Dinvalid.mlir40 {operand_segment_sizes = dense<[0, 1, 1, 1, 1, 1, 0, 0]> : vector<8xi32>}
59 {operand_segment_sizes = dense<[0, 1, 1, 1, 1, 1, 1, 0]> : vector<8xi32>}
/external/llvm-project/mlir/test/Dialect/SCF/
Dinvalid.mlir134 }) {operand_segment_sizes = dense<[1, 1, 1, 0]>: vector<4xi32>}: (index, index, index) -> ()
146 }) {operand_segment_sizes = dense<[1, 1, 1, 0]>: vector<4xi32>}: (index, index, index) -> ()
/external/llvm-project/mlir/include/mlir/Dialect/Linalg/IR/
DLinalgStructuredOpsInterface.td902 // Helper functions to mutate the `operand_segment_sizes` attribute.
911 auto attr = getOperation()->getAttr("operand_segment_sizes")
916 getOperation()->setAttr("operand_segment_sizes", newAttr);
/external/llvm-project/mlir/test/Dialect/SPIRV/
Dcontrol-flow-ops.mlir120 …"spv.BranchConditional"(%true)[^one] {operand_segment_sizes = dense<[1, 0, 0]>: vector<3xi32>} : (…
133 … operand_segment_sizes = dense<[1, 0, 0]>: vector<3xi32>} : (i1) -> ()
/external/tensorflow/tensorflow/compiler/mlir/hlo/tests/
Dhlo-legalize-to-lhlo.mlir601 …d_config = "", call_target_name = "foo", has_side_effect = false, operand_segment_sizes = dense<[2…
613 …d_config = "", call_target_name = "foo", has_side_effect = false, operand_segment_sizes = dense<2>…
/external/llvm-project/mlir/tools/mlir-linalg-ods-gen/
Dmlir-linalg-ods-gen.cpp1471 "operand_segment_sizes",
1492 "operand_segment_sizes",
/external/tensorflow/tensorflow/compiler/mlir/hlo/include/mlir-hlo/Dialect/mhlo/IR/
Dlhlo_ops.td238 // TODO(timshen): Add a custom parser to hide operand_segment_sizes. For example,
/external/tensorflow/tensorflow/compiler/mlir/xla/tests/hlo_to_lhlo_with_xla/
Dhlo_text_to_lhlo_no_opt.hlotxt95 // CHECK-SAME: operand_segment_sizes = dense<[4, 1]> : vector<2xi32>

12