Home
last modified time | relevance | path

Searched refs:SetBufferLayout (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgpu_layout_assignment.cc155 constraints->SetBufferLayout(result_shape.layout(), *call_result_buf)); in AddBackendConstraintsToDnnConvCustomCall()
241 constraints->SetBufferLayout(keys_layout, *output_buffer)); in AddBackendConstraints()
300 TF_RETURN_IF_ERROR(constraints->SetBufferLayout( in PropagateOperandConstraint()
315 TF_RETURN_IF_ERROR(constraints->SetBufferLayout( in PropagateOperandConstraint()
/external/tensorflow/tensorflow/compiler/xla/service/
Dlayout_assignment.cc154 Status LayoutConstraints::SetBufferLayout(const Layout& layout, in SetBufferLayout() function in xla::LayoutConstraints
314 return SetBufferLayout(subshape.layout(), *buffers[0], mandatory); in SetInstructionLayout()
507 constraints->SetBufferLayout(new_shape.layout(), *buffer)); in AddMandatoryConstraints()
1209 TF_RETURN_IF_ERROR(constraints->SetBufferLayout( in PropagateUseConstraintToDefs()
1341 TF_RETURN_IF_ERROR(constraints->SetBufferLayout( in PropagateOperandConstraint()
1367 TF_RETURN_IF_ERROR(constraints->SetBufferLayout( in PropagateOperandConstraint()
1764 TF_RETURN_IF_ERROR(constraints.SetBufferLayout(new_layout, buffer, in RunOnComputation()
Dlayout_assignment.h168 Status SetBufferLayout(const Layout& layout, const LogicalBuffer& buffer,