Home
last modified time | relevance | path

Searched refs:InferArrayLayout (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dlayout_assignment.cc1866 StatusOr<Layout> LayoutAssignment::InferArrayLayout( in InferArrayLayout() function in xla::LayoutAssignment
2006 InferArrayLayout(instruction, index)); in AssignLayouts()
2091 auto param_layout = InferArrayLayout(operand, index); in CalculateComputationLayout()
2314 TF_ASSIGN_OR_RETURN(auto op_layout, InferArrayLayout(instruction, {})); in ConstrainChannelLayouts()
H A Dlayout_assignment.h432 StatusOr<Layout> InferArrayLayout(const HloInstruction* instruction,