| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | gpu_layout_assignment_test.cc | 74 computation->ComputeProgramShape()); in TEST_F()
110 module->entry_computation()->ComputeProgramShape(), in TEST_F()
136 module->entry_computation()->ComputeProgramShape(), in TEST_F()
163 module->entry_computation()->ComputeProgramShape(), in TEST_F()
189 module->entry_computation()->ComputeProgramShape(), in TEST_F()
216 module->entry_computation()->ComputeProgramShape(), in TEST_F()
243 module->entry_computation()->ComputeProgramShape(), in TEST_F()
278 module->entry_computation()->ComputeProgramShape(), in TEST_F()
303 module->entry_computation()->ComputeProgramShape(), in TEST_F()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | layout_assignment_test.cc | 112 ComputationLayout computation_layout(computation->ComputeProgramShape()); in TEST_F()
147 ComputationLayout computation_layout(computation->ComputeProgramShape()); in TEST_F()
194 ComputationLayout computation_layout(computation->ComputeProgramShape()); in TEST_F()
236 m->entry_computation()->ComputeProgramShape()); in TEST_F()
274 m->entry_computation()->ComputeProgramShape()); in TEST_F()
337 ComputationLayout computation_layout(computation->ComputeProgramShape()); in TEST_F()
376 ComputationLayout computation_layout(computation->ComputeProgramShape()); in TEST_F()
414 ComputationLayout computation_layout(computation->ComputeProgramShape()); in TEST_F()
456 ComputationLayout computation_layout(computation->ComputeProgramShape()); in TEST_F()
528 ComputationLayout computation_layout(computation->ComputeProgramShape()); in TEST_F()
[all …]
|
| D | computation_layout.cc | 74 ProgramShape ComputationLayout::ComputeProgramShape() const { in ComputeProgramShape() function in xla::ComputationLayout
|
| D | computation_layout.h | 90 ProgramShape ComputeProgramShape() const;
|
| D | hlo_module.cc | 70 entry_computation_->ComputeProgramShape()); in ReplaceEntryComputation()
84 entry_computation_->ComputeProgramShape()); in AddComputationInternal()
89 entry_computation_->ComputeProgramShape()); in AddComputationInternal()
309 entry_computation_layout().ComputeProgramShape().ToProto(); in ToProto()
465 entry->ComputeProgramShape().result(), proto.input_output_alias())); in CreateFromProto()
|
| D | hlo_verifier.cc | 1007 const ProgramShape& to_apply = instruction.to_apply()->ComputeProgramShape(); in SameElementTypesForOperandsAndToApplyParameters()
1036 reduce->to_apply()->ComputeProgramShape()))); in HandleReduce()
1254 operand_shapes, map->to_apply()->ComputeProgramShape(), map_dims))); in HandleMap()
1272 reduce_window->to_apply()->ComputeProgramShape()))); in HandleReduceWindow()
1285 instruction->select()->ComputeProgramShape(), instruction->window(), in HandleSelectAndScatter()
1287 instruction->scatter()->ComputeProgramShape())); in HandleSelectAndScatter()
1397 async_op->async_wrapped_computation()->ComputeProgramShape(); in CheckAsyncOpComputationShapes()
1670 arg_shapes, scatter->to_apply()->ComputeProgramShape(), in HandleScatter()
|
| D | hlo_module.h | 144 return ComputationLayout(entry_computation()->ComputeProgramShape(), in compute_computation_layout()
|
| D | hlo_computation.cc | 598 ShapeUtil::HumanString(ComputeProgramShape(options.print_ids()))); in ToCord()
647 *proto.mutable_program_shape() = ComputeProgramShape().ToProto(); in ToProto()
878 ProgramShape HloComputation::ComputeProgramShape(bool include_ids) const { in ComputeProgramShape() function in xla::HloComputation
|
| D | hlo_parser.cc | 1828 condition.value()->ComputeProgramShape(), in CreateInstruction()
1829 body.value()->ComputeProgramShape(), operands[0]->shape()); in CreateInstruction()
1941 arg_shapes, to_apply.value()->ComputeProgramShape()); in CreateInstruction()
1969 reduce_computation.value()->ComputeProgramShape()); in CreateInstruction()
2174 arg_shapes, to_apply.value()->ComputeProgramShape(), in CreateInstruction()
2206 reduce_computation.value()->ComputeProgramShape()); in CreateInstruction()
2254 operands[0]->shape(), select.value()->ComputeProgramShape(), in CreateInstruction()
2256 scatter.value()->ComputeProgramShape()); in CreateInstruction()
2564 computation->ComputeProgramShape()); in CreateInstruction()
2896 arg_shapes, update_computation.value()->ComputeProgramShape(), in CreateInstruction()
|
| D | layout_assignment.h | 145 ? ComputationLayout(computation->ComputeProgramShape(), in ComputationLayoutConstraint()
|
| D | hlo_computation.h | 383 ProgramShape ComputeProgramShape(bool include_ids = true) const;
|
| D | hlo_creation_utils.cc | 369 operand_shapes, map_computation->ComputeProgramShape(), map_dims)); in MakeMapHlo()
|
| D | layout_assignment.cc | 1088 branch_comp->ComputeProgramShape(), in CopyOperandIfLayoutsDiffer()
2365 computation->ComputeProgramShape(), in PropagateComputationLayouts()
|
| D | space_to_batch_converter.cc | 2114 new_shape, select_comp->ComputeProgramShape(), new_win, in Propagate()
2116 scatter_comp->ComputeProgramShape())); in Propagate()
|
| D | hlo_evaluator_test.cc | 2944 compute_tuple->ComputeProgramShape())); in TEST_P()
2995 compute_tuple->ComputeProgramShape())); in TEST_P()
|
| D | hlo_evaluator_typed_visitor.h | 1767 /*to_apply_shape=*/function->ComputeProgramShape()));
|
| D | hlo_evaluator.cc | 3722 /*to_apply=*/function->ComputeProgramShape())); in HandleReduce()
|
| D | algebraic_simplifier_test.cc | 5761 add_computation->ComputeProgramShape())); in TEST_P()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | cpu_layout_assignment_test.cc | 78 ComputationLayout computation_layout(computation->ComputeProgramShape()); in TEST_F()
119 ComputationLayout computation_layout(computation->ComputeProgramShape()); in TEST_F()
163 ComputationLayout computation_layout(computation->ComputeProgramShape()); in TEST_F()
197 ComputationLayout computation_layout(computation->ComputeProgramShape()); in TEST_F()
237 ComputationLayout computation_layout(computation->ComputeProgramShape()); in TEST_F()
312 ComputationLayout computation_layout(computation->ComputeProgramShape()); in RunDotOutputFusion()
456 ComputationLayout computation_layout(computation->ComputeProgramShape()); in TEST_F()
|
| /external/tensorflow/tensorflow/compiler/xrt/kernels/ |
| D | xrt_compile_ops.cc | 234 .ComputeProgramShape() in Compute()
|
| /external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
| D | spmd_partitioner.cc | 3516 hlo->to_apply()->ComputeProgramShape())); in HandleReduce()
3928 hlo->to_apply()->ComputeProgramShape())); in HandleReduceWindow()
4508 auto program_shape = module->entry_computation()->ComputeProgramShape(); in Run()
4521 auto new_program_shape = module->entry_computation()->ComputeProgramShape(); in Run()
|
| D | dot_handler.cc | 4346 reduce_outside->to_apply()->ComputeProgramShape())); in MoveUsersIntoWindowedDotGeneralLoopOnNonContractingDimensions()
|
| /external/tensorflow/tensorflow/compiler/xrt/tests/ |
| D | raw_api_test.cc | 381 ->ComputeProgramShape(); in XlaCompiledProgramShape()
|