Searched refs:infeed_shape (Results 1 – 11 of 11) sorted by relevance
| /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/core/tpu/kernels/xla/ |
| D | infeed_op.cc | 42 xla::Shape infeed_shape = ApiConverter::FromC(&c_infeed_shape); in GetTPUInfeedLayout() local
45 return infeed_shape; in GetTPUInfeedLayout()
|
| /external/tensorflow/tensorflow/compiler/xla/tools/ |
| D | replay_computation.cc | 265 if (absl::optional<Shape> infeed_shape = GetXfeedShape( in ReplayComputation() local
268 std::move(MakeFakeLiteral(*infeed_shape)).ValueOrDie()); in ReplayComputation()
|
| /external/tensorflow/tensorflow/core/tpu/kernels/ |
| D | infeed_ops.cc | 65 xla::Shape infeed_shape = ApiConverter::FromC(&c_infeed_shape); in GetTPUInfeedLayout() local
68 return infeed_shape; in GetTPUInfeedLayout()
|
| /external/tensorflow/tensorflow/stream_executor/tpu/ |
| D | tpu_executor_c_api.h | 205 XLA_Shape* infeed_shape);
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_instructions.h | 1156 explicit HloInfeedInstruction(const Shape& infeed_shape,
1167 const Shape& infeed_shape() const { in infeed_shape() function
|
| D | hlo_instructions.cc | 2047 HloInfeedInstruction::HloInfeedInstruction(const Shape& infeed_shape, in HloInfeedInstruction() argument
2052 {infeed_shape, ShapeUtil::MakeTokenShape()})), in HloInfeedInstruction()
2084 infeed_shape(), new_operands[0], infeed_config()); in CloneWithNewOperandsImpl()
|
| D | hlo_instruction.h | 736 const Shape& infeed_shape, HloInstruction* token_operand,
|
| D | hlo_verifier.cc | 428 {infeed->infeed_shape(), ShapeUtil::MakeTokenShape()})); in HandleInfeed()
|
| D | hlo_instruction.cc | 1109 const Shape& infeed_shape, HloInstruction* token_operand, in CreateInfeed() argument
1111 return absl::make_unique<HloInfeedInstruction>(infeed_shape, token_operand, in CreateInfeed()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | ir_emitter.cc | 395 const Shape& data_shape = infeed->infeed_shape(); in HandleInfeed()
|