Home
last modified time | relevance | path

Searched refs:kOperandSegmentSizesAttr (Results 1 – 2 of 2) sorted by relevance

/external/llvm-project/mlir/lib/Dialect/Async/IR/
DAsync.cpp121 constexpr char kOperandSegmentSizesAttr[] = "operand_segment_sizes"; variable
157 result.addAttribute(kOperandSegmentSizesAttr, operandSegmentSizes); in build()
208 p.printOptionalAttrDictWithKeyword(op.getAttrs(), {kOperandSegmentSizesAttr}); in print()
275 result.addAttribute(kOperandSegmentSizesAttr, operandSegmentSizes); in parseExecuteOp()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/
Dtf_device.cc320 constexpr char kOperandSegmentSizesAttr[] = "operand_segment_sizes"; variable
343 if (!state->attributes.get(kOperandSegmentSizesAttr)) { in ParseReplicateOp()
349 state->addAttribute(kOperandSegmentSizesAttr, attr); in ParseReplicateOp()
409 kOperandSegmentSizesAttr}); in Print()
550 state->addAttribute(kOperandSegmentSizesAttr, operand_segment_sizes); in BuildReplicateOp()