HomeSort by relevance Sort by last modified time
    Searched full:output_shape_with_layout (Results 1 - 4 of 4) sorted by null

  /external/tensorflow/tensorflow/compiler/xla/
shape_util.h 535 // ShapeUtil::ReshapeIsBitcast(input_shape, output_shape_with_layout) returns
536 // true (where 'output_shape_with_layout' is 'output_shape' with the found
538 // 'output_shape_with_layout' if such a layout can be found, and an error
shape_util.cc     [all...]
  /external/tensorflow/tensorflow/compiler/xla/service/
layout_assignment_test.cc 417 Shape output_shape_with_layout(cshape);
419 *output_shape_with_layout.mutable_layout() =
426 ShapeLayout(output_shape_with_layout);
    [all...]
layout_assignment.cc 924 Shape output_shape_with_layout = ShapeUtil::MakeShapeWithLayout( local
    [all...]

Completed in 170 milliseconds