Searched refs:infeed_shape (Results 1 – 9 of 9) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | infeed_thunk.cc | 40 const auto& infeed_shape = infeed_slices_.shape(); in ExecuteOnStream() local
41 TF_RET_CHECK(infeed_shape.IsTuple()) in ExecuteOnStream()
42 << ShapeUtil::HumanStringWithLayout(infeed_shape); in ExecuteOnStream()
43 TF_RET_CHECK(infeed_shape.tuple_shapes().size() == 2) in ExecuteOnStream()
44 << ShapeUtil::HumanStringWithLayout(infeed_shape); in ExecuteOnStream()
45 TF_RET_CHECK(infeed_shape.tuple_shapes(1).IsToken()) in ExecuteOnStream()
46 << ShapeUtil::HumanStringWithLayout(infeed_shape); in ExecuteOnStream()
48 ShapeUtil::Equal(infeed_buffers.shape(), infeed_shape.tuple_shapes(0))) in ExecuteOnStream()
50 << ShapeUtil::HumanStringWithLayout(infeed_shape.tuple_shapes(0)) in ExecuteOnStream()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/tests/ |
| D | cpu_infeed_test.cc | 121 const auto infeed_shape = ShapeUtil::MakeShape(F32, {3}); in TEST_F() local
138 auto infeed = Infeed(&builder, infeed_shape); in TEST_F()
223 const auto build_body = [&result_shape](const Shape& infeed_shape) { in TEST_F() argument
227 auto infeed = Infeed(&builder, infeed_shape); in TEST_F()
|
| /external/tensorflow/tensorflow/compiler/xla/tools/ |
| D | replay_computation.cc | 233 if (absl::optional<Shape> infeed_shape = GetXfeedShape( in ReplayComputation() local
236 std::move(MakeFakeLiteral(*infeed_shape)).ValueOrDie()); in ReplayComputation()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_instructions.h | 964 explicit HloInfeedInstruction(const Shape& infeed_shape,
975 const Shape& infeed_shape() const { in infeed_shape() function
|
| D | hlo_instructions.cc | 1772 HloInfeedInstruction::HloInfeedInstruction(const Shape& infeed_shape, in HloInfeedInstruction() argument
1777 {infeed_shape, ShapeUtil::MakeTokenShape()})), in HloInfeedInstruction()
1809 infeed_shape(), new_operands[0], infeed_config()); in CloneWithNewOperandsImpl()
|
| D | hlo_instruction.h | 537 const Shape& infeed_shape, HloInstruction* token_operand,
|
| D | hlo_verifier.cc | 253 {infeed->infeed_shape(), ShapeUtil::MakeTokenShape()})); in HandleInfeed()
|
| D | hlo_instruction.cc | 863 const Shape& infeed_shape, HloInstruction* token_operand, in CreateInfeed() argument
865 return absl::make_unique<HloInfeedInstruction>(infeed_shape, token_operand, in CreateInfeed()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | ir_emitter.cc | 331 const Shape& data_shape = infeed->infeed_shape(); in HandleInfeed()
|