Home
last modified time | relevance | path

Searched refs:AddBackendConstraints (Results 1 – 6 of 6) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/cpu/
H A Dcpu_layout_assignment.h40 Status AddBackendConstraints(LayoutConstraints* constraints) override;
H A Dcpu_layout_assignment.cc110 Status CpuLayoutAssignment::AddBackendConstraints( in AddBackendConstraints() function in xla::cpu::CpuLayoutAssignment
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/
H A Dgpu_layout_assignment.h41 Status AddBackendConstraints(LayoutConstraints* constraints) override;
H A Dgpu_layout_assignment.cc244 Status GpuLayoutAssignment::AddBackendConstraints( in AddBackendConstraints() function in xla::gpu::GpuLayoutAssignment
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dlayout_assignment.h496 virtual Status AddBackendConstraints(LayoutConstraints* constraints) { in AddBackendConstraints() function
H A Dlayout_assignment.cc2255 TF_RETURN_IF_ERROR(AddBackendConstraints(constraints)); in RunOnComputation()