Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgpu_layout_assignment.cc230 Shape op1_shape = instruction->operand(1)->shape(); in AddBackendConstraints() local
231 LayoutUtil::SetToDefaultLayout(&op1_shape); in AddBackendConstraints()
237 constraints->SetOperandLayout(op1_shape, instruction, 1)); in AddBackendConstraints()
291 Shape op1_shape = instruction->operand(1)->shape(); in AddBackendConstraints() local
294 set_fortran_layout(&op1_shape); in AddBackendConstraints()
299 constraints->SetOperandLayout(op1_shape, instruction, 1)); in AddBackendConstraints()