Searched refs:ComputeProgramShape (Results 1 – 16 of 16) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | layout_assignment_test.cc | 111 ComputationLayout computation_layout(computation->ComputeProgramShape()); in TEST_F()
146 ComputationLayout computation_layout(computation->ComputeProgramShape()); in TEST_F()
193 ComputationLayout computation_layout(computation->ComputeProgramShape()); in TEST_F()
235 m->entry_computation()->ComputeProgramShape()); in TEST_F()
273 m->entry_computation()->ComputeProgramShape()); in TEST_F()
308 m->entry_computation()->ComputeProgramShape()); in TEST_F()
371 ComputationLayout computation_layout(computation->ComputeProgramShape()); in TEST_F()
412 ComputationLayout computation_layout(computation->ComputeProgramShape()); in TEST_F()
450 ComputationLayout computation_layout(computation->ComputeProgramShape()); in TEST_F()
492 ComputationLayout computation_layout(computation->ComputeProgramShape()); in TEST_F()
[all …]
|
| D | computation_layout.h | 88 ProgramShape ComputeProgramShape() const;
|
| D | computation_layout.cc | 59 ProgramShape ComputationLayout::ComputeProgramShape() const { in ComputeProgramShape() function in xla::ComputationLayout
|
| D | hlo_verifier.cc | 426 const ProgramShape& to_apply = instruction.to_apply()->ComputeProgramShape(); in SameElementTypesForOperandsAndToApplyParameters()
455 reduce->to_apply()->ComputeProgramShape()))); in HandleReduce()
614 operand_shapes, map->to_apply()->ComputeProgramShape(), map_dims))); in HandleMap()
628 reduce_window->to_apply()->ComputeProgramShape()))); in HandleReduceWindow()
641 instruction->select()->ComputeProgramShape(), instruction->window(), in HandleSelectAndScatter()
643 instruction->scatter()->ComputeProgramShape())); in HandleSelectAndScatter()
823 scatter->to_apply()->ComputeProgramShape(), in HandleScatter()
|
| D | hlo_module.cc | 62 entry_computation_->ComputeProgramShape()); in AddComputationInternal()
237 entry_computation_layout().ComputeProgramShape().ToProto(); in ToProto()
347 entry->ComputeProgramShape().result(), proto.input_output_alias())); in CreateFromProto()
|
| D | hlo_module.h | 123 return ComputationLayout(entry_computation()->ComputeProgramShape(), in compute_computation_layout()
|
| D | hlo_computation.cc | 490 s << ShapeUtil::HumanString(ComputeProgramShape()) << " "; in ToString()
530 *proto.mutable_program_shape() = ComputeProgramShape().ToProto(); in ToProto()
707 ProgramShape HloComputation::ComputeProgramShape() const { in ComputeProgramShape() function in xla::HloComputation
|
| D | hlo_computation.h | 270 ProgramShape ComputeProgramShape() const;
|
| D | hlo_creation_utils.cc | 245 operand_shapes, map_computation->ComputeProgramShape(), map_dims)); in MakeMapHlo()
|
| D | layout_assignment.cc | 1667 ComputationLayout computation_layout(computation->ComputeProgramShape(), in CalculateComputationLayout()
1892 computation->ComputeProgramShape(), in PropagateComputationLayouts()
|
| D | hlo_evaluator_typed_visitor.h | 1699 /*to_apply=*/function->ComputeProgramShape())); in HandleReduce()
1970 /*to_apply_shape=*/function->ComputeProgramShape())); in HandleReduceWindow()
|
| D | algebraic_simplifier_test.cc | 4264 add_computation->ComputeProgramShape())); in TEST_P()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | gpu_layout_assignment_test.cc | 69 computation->ComputeProgramShape()); in TEST_F()
155 ComputationLayout computation_layout(computation->ComputeProgramShape()); in TEST_F()
224 ComputationLayout computation_layout(computation->ComputeProgramShape()); in TEST_F()
306 computation->ComputeProgramShape()); in TEST_F()
354 module->entry_computation()->ComputeProgramShape(), in TEST_F()
392 module->entry_computation()->ComputeProgramShape(), in TEST_F()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | cpu_layout_assignment_test.cc | 79 ComputationLayout computation_layout(computation->ComputeProgramShape()); in TEST_F()
120 ComputationLayout computation_layout(computation->ComputeProgramShape()); in TEST_F()
164 ComputationLayout computation_layout(computation->ComputeProgramShape()); in TEST_F()
198 ComputationLayout computation_layout(computation->ComputeProgramShape()); in TEST_F()
238 ComputationLayout computation_layout(computation->ComputeProgramShape()); in TEST_F()
306 ComputationLayout computation_layout(computation->ComputeProgramShape()); in RunDotOutputFusion()
|
| /external/tensorflow/tensorflow/compiler/xrt/kernels/ |
| D | xrt_compile_ops.cc | 191 .ComputeProgramShape() in Compute()
|
| /external/tensorflow/tensorflow/compiler/xrt/tests/ |
| D | raw_api_test.cc | 243 ->ComputeProgramShape(); in XlaCompiledProgramShape()
|