| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| thunk.cc | 43 case Thunk::kInfeed:
44 return "kInfeed";
|
| thunk.h | 55 kInfeed,
|
| infeed_thunk.cc | 28 : Thunk(Kind::kInfeed, hlo_instruction), infeed_slices_(infeed_slices) {}
|
| ir_emitter_unnested.cc | [all...] |
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| hlo_opcode.h | 94 V(kInfeed, "infeed", 1) \
|
| hlo_element_type_converter.cc | 62 // non-nested tuple. This assumption is currently safe as only kTuple, kInfeed,
132 opcode == HloOpcode::kInfeed || opcode == HloOpcode::kOutfeed) {
|
| instruction_fusion.cc | 79 case HloOpcode::kInfeed:
|
| layout_assignment.cc | 443 if (instruction->opcode() == HloOpcode::kInfeed) {
[all...] |
| hlo_instruction.cc | 358 case HloOpcode::kInfeed: {
[all...] |
| hlo_verifier.cc | 770 case HloOpcode::kInfeed:
[all...] |
| hlo_graph_dumper.cc | [all...] |
| hlo_instructions.cc | [all...] |
| hlo_parser.cc | [all...] |
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| parallel_task_assignment.cc | 143 opcode == HloOpcode::kFft || opcode == HloOpcode::kInfeed ||
|
| ir_emitter.h | 548 kInfeed,
|
| ir_emitter.cc | 369 XfeedKind::kInfeed, tuple_element_shape, tuple_element_address));
378 EmitXfeedTransfer(XfeedKind::kInfeed, data_shape, data_address));
409 if (kind == XfeedKind::kInfeed) {
433 if (kind == XfeedKind::kInfeed) {
458 if (kind == XfeedKind::kInfeed) {
[all...] |
| /external/tensorflow/tensorflow/compiler/xla/tools/ |
| replay_computation.cc | 117 ? HloOpcode::kInfeed
158 // kInfeed instructions should have a shape (buffer, token). kOutfeed
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| xla_builder.cc | 229 case HloOpcode::kInfeed:
[all...] |
