Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgpu_layout_assignment.cc228 Shape op0_shape = instruction->operand(0)->shape(); in AddBackendConstraints() local
229 LayoutUtil::SetToDefaultLayout(&op0_shape); in AddBackendConstraints()
235 constraints->SetOperandLayout(op0_shape, instruction, 0)); in AddBackendConstraints()
242 Shape op0_shape = instruction->operand(0)->shape(); in AddBackendConstraints() local
243 LayoutUtil::SetToDefaultLayout(&op0_shape); in AddBackendConstraints()
247 constraints->SetOperandLayout(op0_shape, instruction, 0)); in AddBackendConstraints()
290 Shape op0_shape = instruction->operand(0)->shape(); in AddBackendConstraints() local
293 set_fortran_layout(&op0_shape); in AddBackendConstraints()
297 constraints->SetOperandLayout(op0_shape, instruction, 0)); in AddBackendConstraints()