Searched refs:infeed_config (Results 1 – 15 of 15) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_memory_scheduler.cc | 223 if (opcode == HloOpcode::kInfeed && !instruction->infeed_config().empty()) { in BytesFreedIfScheduled()
|
| D | hlo.proto | 135 bytes infeed_config = 27; field
|
| D | hlo_instruction.h | 1795 string infeed_config() const;
|
| D | hlo_instruction.cc | 402 CreateInfeed(data_shape, operands(0), proto.infeed_config()); in CreateFromProto()
4103 string HloInstruction::infeed_config() const { in infeed_config() function in xla::HloInstruction
4104 return Cast<HloInfeedInstruction>(this)->infeed_config(); in infeed_config()
|
| D | hlo_instructions.h | 1162 string infeed_config() const { return infeed_config_; } in infeed_config() function
|
| D | hlo_instructions.cc | 2084 infeed_shape(), new_operands[0], infeed_config()); in CloneWithNewOperandsImpl()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/tests/translate/ |
| D | export.mlir | 529 …%0 = "mhlo.infeed"(%arg0) {infeed_config = "foobar", layout=[[[0,1], [0]], unit]} : (!mhlo.token) …
535 …ROOT %[[RESULT:.*]] = ((s32[3,3], pred[]), token[]) infeed(token[] [[ARG]]), infeed_config="foobar"
|
| D | import.hlotxt | 453 // CHECK-SAME: infeed_config = "foobar"
454 ROOT %infeed = (s32[3], token[]) infeed(token[] %token0), infeed_config="foobar"
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/ |
| D | hlo_function_importer.cc | 385 instruction->infeed_config()))); in ImportInstructionImpl()
|
| D | mlir_hlo_to_hlo.cc | 781 std::string(op.infeed_config())); in ExportXlaOp()
|
| /external/tensorflow/tensorflow/compiler/mlir/hlo/tests/ |
| D | ops.mlir | 469 …%0 = "mhlo.infeed"(%token) {infeed_config = "foobar", layout = [[[0]], unit, [0]]} : (!mhlo.token)…
477 …%0 = "mhlo.infeed"(%token) {infeed_config = "foobar", layout = [[[0]], [0]]} : (!mhlo.token) -> tu…
|
| /external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
| D | spmd_partitioner.cc | 2502 HloInstruction::CreateInfeed(shape, token, hlo->infeed_config())); in HandleInfeed()
2511 shard_shape, token, hlo->infeed_config())); in HandleInfeed()
2571 per_branch_partitioned_shapes[i], param, hlo->infeed_config())); in HandleInfeed()
|
| /external/tensorflow/tensorflow/compiler/mlir/hlo/include/mlir-hlo/Dialect/mhlo/IR/ |
| D | hlo_ops.td | 414 DefaultValuedAttr<StrAttr, "">:$infeed_config,
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | mhlo_to_lhlo_with_xla.cc | 1113 infeed_op.configAttr(builder_.getStringAttr(infeed->infeed_config())); in EmitInfeedOp()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/tests/ |
| D | legalize-tf.mlir | 1078 // CHECK: [[INFEED:%.*]] = "mhlo.infeed"([[TOKEN]]) {infeed_config = ""} : (!mhlo.token) -> tuple<t…
|