| /external/tensorflow/tensorflow/core/tpu/kernels/xla/ |
| D | outfeed_ops.cc | 41 const string outfeed_config; in Compile() local
42 xla::Outfeed(ctx->Input(0), xla_shape, outfeed_config); in Compile()
78 const string outfeed_config; in Compile() local
79 xla::Outfeed(tuple, tuple_shape, outfeed_config); in Compile()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/ir/ |
| D | mlir_hlo_builder.h | 232 const string& outfeed_config) override;
|
| D | mlir_hlo_builder.cc | 481 const string& outfeed_config) { in OutfeedWithTokenInternal() argument
484 loc_, token_type, GetValue(operand), GetValue(token), outfeed_config)); in OutfeedWithTokenInternal()
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.h | 642 const string& outfeed_config);
645 const string& outfeed_config);
648 const string& outfeed_config);
1212 const string& outfeed_config);
1424 const string& outfeed_config);
2003 const string& outfeed_config);
2010 const string& outfeed_config);
|
| D | xla_builder.cc | 1769 const string& outfeed_config) { in Outfeed() argument
1788 instr.set_outfeed_config(outfeed_config); in Outfeed()
1822 const string& outfeed_config) { in OutfeedWithToken() argument
1836 outfeed_config); in OutfeedWithToken()
1842 const string& outfeed_config) { in OutfeedWithTokenInternal() argument
1846 instr.set_outfeed_config(outfeed_config); in OutfeedWithTokenInternal()
4454 const string& outfeed_config) { in Outfeed() argument
4455 return operand.builder()->Outfeed(operand, shape_with_layout, outfeed_config); in Outfeed()
4928 const string& outfeed_config) { in OutfeedWithToken() argument
4930 outfeed_config); in OutfeedWithToken()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_instruction.h | 744 HloInstruction* token_operand, absl::string_view outfeed_config);
1801 const string& outfeed_config() const;
|
| D | hlo_memory_scheduler.cc | 220 !instruction->outfeed_config().empty()) { in BytesFreedIfScheduled()
|
| D | hlo_instructions.cc | 2090 absl::string_view outfeed_config) in HloOutfeedInstruction() argument
2093 outfeed_config_(outfeed_config) { in HloOutfeedInstruction()
2100 proto.set_outfeed_config(outfeed_config()); in ToProto()
2130 outfeed_shape(), new_operands[0], new_operands[1], outfeed_config()); in CloneWithNewOperandsImpl()
|
| D | hlo_instruction.cc | 409 proto.outfeed_config()); in CreateFromProto()
1117 HloInstruction* token_operand, absl::string_view outfeed_config) { in CreateOutfeed() argument
1119 outfeed_shape, operand, token_operand, outfeed_config); in CreateOutfeed()
4119 const string& HloInstruction::outfeed_config() const { in outfeed_config() function in xla::HloInstruction
4120 return Cast<HloOutfeedInstruction>(this)->outfeed_config(); in outfeed_config()
|
| D | hlo_instructions.h | 1195 absl::string_view outfeed_config);
1201 const string& outfeed_config() const { return outfeed_config_; } in outfeed_config() function
|
| D | hlo.proto | 115 bytes outfeed_config = 22; field
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/tests/translate/ |
| D | import.hlotxt | 566 // CHECK-SAME: outfeed_config = "foobar"
567 ROOT %outfeed.3 = token[] outfeed(s32[3] %Arg_0.1, token[] %Arg_1.2), outfeed_config="foobar"
|
| D | export.mlir | 579 …%0 = "mhlo.outfeed"(%data, %token) {outfeed_config = "foobar"} : (tensor<3xi32>, !mhlo.token) -> !…
586 …ken[] outfeed(s32[3] [[DATA]], token[] [[TOKEN]]), outfeed_shape=s32[3]{0}, outfeed_config="foobar"
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/ |
| D | hlo_function_importer.cc | 405 instruction->outfeed_config()))); in ImportInstructionImpl()
|
| D | mlir_hlo_to_hlo.cc | 812 std::string(op.outfeed_config())); in ExportXlaOp()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | mhlo_to_lhlo_with_xla.cc | 1127 outfeed_op.configAttr(builder_.getStringAttr(outfeed->outfeed_config())); in EmitOutfeedOp()
|
| /external/tensorflow/tensorflow/compiler/mlir/hlo/include/mlir-hlo/Dialect/mhlo/IR/ |
| D | hlo_ops.td | 438 DefaultValuedAttr<StrAttr, "">:$outfeed_config
|
| /external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
| D | spmd_partitioner.cc | 2917 outfeed_shape, operand, token, hlo->outfeed_config())); in HandleOutfeed()
3030 hlo->outfeed_config())); in HandleOutfeed()
|
| /external/tensorflow/tensorflow/compiler/mlir/hlo/tests/ |
| D | canonicalize.mlir | 636 %2 = "mhlo.outfeed"(%arg1, %1) {outfeed_config = ""} : (tensor<i64>, !mhlo.token) -> !mhlo.token
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/tests/ |
| D | legalize-tf.mlir | 1460 // CHECK: "mhlo.outfeed"([[TUPLE]], [[TOKEN]]) {outfeed_config = ""} : (tuple<tensor<3xi32>, tensor…
|