| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | gpu_layout_assignment_test.cc | 71 computation->ComputeProgramShape()); in TEST_F()
157 ComputationLayout computation_layout(computation->ComputeProgramShape()); in TEST_F()
226 ComputationLayout computation_layout(computation->ComputeProgramShape()); in TEST_F()
308 computation->ComputeProgramShape()); in TEST_F()
356 module->entry_computation()->ComputeProgramShape(), in TEST_F()
394 module->entry_computation()->ComputeProgramShape(), in TEST_F()
421 module->entry_computation()->ComputeProgramShape(), in TEST_F()
|
| /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 | 90 ProgramShape ComputeProgramShape() const;
|
| D | hlo_module.cc | 62 entry_computation_->ComputeProgramShape()); in ReplaceEntryComputation()
76 entry_computation_->ComputeProgramShape()); in AddComputationInternal()
81 entry_computation_->ComputeProgramShape()); in AddComputationInternal()
285 entry_computation_layout().ComputeProgramShape().ToProto(); in ToProto()
407 entry->ComputeProgramShape().result(), proto.input_output_alias())); in CreateFromProto()
|
| D | computation_layout.cc | 73 ProgramShape ComputationLayout::ComputeProgramShape() const { in ComputeProgramShape() function in xla::ComputationLayout
|
| D | hlo_verifier.cc | 644 const ProgramShape& to_apply = instruction.to_apply()->ComputeProgramShape(); in SameElementTypesForOperandsAndToApplyParameters()
673 reduce->to_apply()->ComputeProgramShape()))); in HandleReduce()
884 operand_shapes, map->to_apply()->ComputeProgramShape(), map_dims))); in HandleMap()
902 reduce_window->to_apply()->ComputeProgramShape()))); in HandleReduceWindow()
915 instruction->select()->ComputeProgramShape(), instruction->window(), in HandleSelectAndScatter()
917 instruction->scatter()->ComputeProgramShape())); in HandleSelectAndScatter()
1144 scatter->to_apply()->ComputeProgramShape(), in HandleScatter()
|
| D | hlo_module.h | 140 return ComputationLayout(entry_computation()->ComputeProgramShape(), in compute_computation_layout()
|
| D | hlo_computation.cc | 565 s << ShapeUtil::HumanString(ComputeProgramShape(options.print_ids())) in ToString()
657 *proto.mutable_program_shape() = ComputeProgramShape().ToProto(); in ToProto()
835 ProgramShape HloComputation::ComputeProgramShape(bool include_ids) const { in ComputeProgramShape() function in xla::HloComputation
|
| D | hlo_parser.cc | 1444 condition.value()->ComputeProgramShape(), in ParseInstructionRhs()
1445 body.value()->ComputeProgramShape(), operands[0]->shape()); in ParseInstructionRhs()
1561 arg_shapes, to_apply.value()->ComputeProgramShape()); in ParseInstructionRhs()
1591 reduce_computation.value()->ComputeProgramShape()); in ParseInstructionRhs()
1803 arg_shapes, to_apply.value()->ComputeProgramShape(), in ParseInstructionRhs()
1838 reduce_computation.value()->ComputeProgramShape()); in ParseInstructionRhs()
1890 operands[0]->shape(), select.value()->ComputeProgramShape(), in ParseInstructionRhs()
1892 scatter.value()->ComputeProgramShape()); in ParseInstructionRhs()
2216 computation->ComputeProgramShape()); in ParseInstructionRhs()
2527 update_computation.value()->ComputeProgramShape(), in ParseInstructionRhs()
|
| D | hlo_computation.h | 311 ProgramShape ComputeProgramShape(bool include_ids = true) const;
|
| D | hlo_creation_utils.cc | 317 operand_shapes, map_computation->ComputeProgramShape(), map_dims)); in MakeMapHlo()
|
| D | layout_assignment.cc | 1808 ComputationLayout computation_layout(computation->ComputeProgramShape(), in CalculateComputationLayout()
2049 computation->ComputeProgramShape(), in PropagateComputationLayouts()
|
| D | space_to_batch_converter.cc | 1477 new_shape, select_comp->ComputeProgramShape(), new_win, in Propagate()
1479 scatter_comp->ComputeProgramShape())); in Propagate()
|
| D | hlo_evaluator_test.cc | 2906 compute_tuple->ComputeProgramShape())); in TEST_P()
2957 compute_tuple->ComputeProgramShape())); in TEST_P()
|
| D | hlo_evaluator.cc | 2417 /*to_apply=*/function->ComputeProgramShape())); in HandleReduce()
|
| D | hlo_evaluator_typed_visitor.h | 1954 /*to_apply_shape=*/function->ComputeProgramShape()));
|
| D | algebraic_simplifier_test.cc | 5121 add_computation->ComputeProgramShape())); in TEST_P()
|
| /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()
313 ComputationLayout computation_layout(computation->ComputeProgramShape()); in RunDotOutputFusion()
458 ComputationLayout computation_layout(computation->ComputeProgramShape()); in TEST_F()
|
| /external/tensorflow/tensorflow/compiler/xrt/kernels/ |
| D | xrt_compile_ops.cc | 233 .ComputeProgramShape() in Compute()
|
| /external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
| D | spmd_partitioner.cc | 2744 hlo->to_apply()->ComputeProgramShape())); in HandleReduce()
3126 hlo->to_apply()->ComputeProgramShape())); in HandleReduceWindow()
3673 auto program_shape = module->entry_computation()->ComputeProgramShape(); in Run()
3686 auto new_program_shape = module->entry_computation()->ComputeProgramShape(); in Run()
|
| D | dot_handler.cc | 3300 reduce_outside->to_apply()->ComputeProgramShape())); in MoveUsersIntoWindowedDotGeneralLoopOnNonContractingDimensions()
|
| /external/tensorflow/tensorflow/compiler/xrt/tests/ |
| D | raw_api_test.cc | 382 ->ComputeProgramShape(); in XlaCompiledProgramShape()
|