| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/ |
| H A D | layout_assignment_test.cc | 112 ComputationLayout computation_layout(computation->ComputeProgramShape()); in TEST_F() local
147 ComputationLayout computation_layout(computation->ComputeProgramShape()); in TEST_F() local
194 ComputationLayout computation_layout(computation->ComputeProgramShape()); in TEST_F() local
235 ComputationLayout computation_layout( in TEST_F() local
273 ComputationLayout computation_layout( in TEST_F() local
337 ComputationLayout computation_layout(computation->ComputeProgramShape()); in TEST_F() local
376 ComputationLayout computation_layout(computation->ComputeProgramShape()); in TEST_F() local
414 ComputationLayout computation_layout(computation->ComputeProgramShape()); in TEST_F() local
456 ComputationLayout computation_layout(computation->ComputeProgramShape()); in TEST_F() local
528 ComputationLayout computation_layout(computation->ComputeProgramShape()); in TEST_F() local
[all …]
|
| H A D | layout_assignment.h | 137 ComputationLayout* computation_layout, in ComputationLayoutConstraint()
149 const ComputationLayout& computation_layout() const { in computation_layout() function
287 const ComputationLayout& computation_layout() const { in computation_layout() function
|
| H A D | layout_assignment.cc | 131 HloComputation* computation, ComputationLayout* computation_layout, in LayoutConstraints()
830 const ComputationLayout& computation_layout) { in CheckCallLayout()
926 const ComputationLayout& computation_layout) { in CheckParameterLayout()
2363 HloComputation* computation, ComputationLayout* computation_layout) { in PropagateComputationLayouts()
|
| H A D | computation_layout.h | 96 friend H AbslHashValue(H h, const ComputationLayout& computation_layout) { in AbslHashValue()
|
| H A D | hlo_module_util.cc | 50 ComputationLayout* computation_layout = in CreateModuleConfig() local
|
| H A D | hlo_parser.cc | 781 ComputationLayout* computation_layout) { in ParseComputationLayout()
4359 ComputationLayout computation_layout(ShapeLayout(Shape{})); in ParseAttributeHelper() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| H A D | gpu_layout_assignment_test.cc | 73 ComputationLayout computation_layout( in TEST_F() local
109 ComputationLayout computation_layout( in TEST_F() local
135 ComputationLayout computation_layout( in TEST_F() local
162 ComputationLayout computation_layout( in TEST_F() local
188 ComputationLayout computation_layout( in TEST_F() local
215 ComputationLayout computation_layout( in TEST_F() local
242 ComputationLayout computation_layout( in TEST_F() local
277 ComputationLayout computation_layout( in TEST_F() local
302 ComputationLayout computation_layout( in TEST_F() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| H A D | cpu_layout_assignment_test.cc | 78 ComputationLayout computation_layout(computation->ComputeProgramShape()); in TEST_F() local
119 ComputationLayout computation_layout(computation->ComputeProgramShape()); in TEST_F() local
163 ComputationLayout computation_layout(computation->ComputeProgramShape()); in TEST_F() local
197 ComputationLayout computation_layout(computation->ComputeProgramShape()); in TEST_F() local
237 ComputationLayout computation_layout(computation->ComputeProgramShape()); in TEST_F() local
312 ComputationLayout computation_layout(computation->ComputeProgramShape()); in RunDotOutputFusion() local
456 ComputationLayout computation_layout(computation->ComputeProgramShape()); in TEST_F() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xrt/kernels/ |
| H A D | xrt_execute_op.cc | 78 const xla::ComputationLayout& computation_layout) { in GetDynamicInputInfo()
92 const xla::ComputationLayout& computation_layout = in GetInputTuples() local
303 const xla::ComputationLayout& computation_layout = in RunExecutable() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/tpu/kernels/ |
| H A D | tpu_compile_op_support.cc | 67 ComputationLayout* computation_layout = in CreateModuleConfig() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/tpu/ |
| H A D | tpu_execute.cc | 456 xla::ComputationLayout computation_layout( in TPUExecute() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/client/ |
| H A D | local_client.cc | 122 const ComputationLayout& computation_layout = in RunHelper() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/pjrt/ |
| H A D | tfrt_cpu_pjrt_client.cc | 1185 const auto& computation_layout = in TfrtCpuExecutable() local
|
| H A D | pjrt_stream_executor_client.cc | 1778 const auto& computation_layout = in PjRtStreamExecutorExecutable() local
|