Searched defs:to_apply (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | shape_inference_test.cc | 63 ProgramShape to_apply = ShapeUtil::MakeProgramShape({f32_, f32_}, f32_); in ExpectInferredReduceShape() local
317 ProgramShape to_apply = ShapeUtil::MakeProgramShape( in TEST_F() local
563 ProgramShape to_apply = ShapeUtil::MakeProgramShape({f32_}, s32_); in TEST_F() local
634 ProgramShape to_apply = ShapeUtil::MakeProgramShape({f32_}, f32_); in TEST_F() local
710 ProgramShape to_apply = ShapeUtil::MakeProgramShape({f32_, f32_}, f32_); in TEST_F() local
720 ProgramShape to_apply = ShapeUtil::MakeProgramShape({f32_, f32_, f32_}, f32_); in TEST_F() local
730 ProgramShape to_apply = ShapeUtil::MakeProgramShape({f32_, f32_}, s32_); in TEST_F() local
|
| D | shape_inference.cc | 1082 const ProgramShape& to_apply, in InferMapShape()
1848 const ProgramShape& to_apply) { in InferReduceShape()
2419 const ProgramShape& to_apply) { in InferCallShape()
|
| D | hlo_instruction.cc | 1965 HloComputation* HloInstruction::to_apply() const { in to_apply() function in xla::HloInstruction
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | xla_data.proto | 539 ComputationHandle to_apply = 2; field
582 ComputationHandle to_apply = 3; field
603 ComputationHandle to_apply = 5; field
610 ComputationHandle to_apply = 5; field
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | cpu_compiler.cc | 575 HloComputation* to_apply = instruction->to_apply(); in RunBackend() local
|
| /external/tensorflow/tensorflow/compiler/xla/tools/parser/ |
| D | hlo_parser.cc | 656 optional<HloComputation*> to_apply; in ParseInstruction() local
740 optional<HloComputation*> to_apply; in ParseInstruction() local
|