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

  /external/tensorflow/tensorflow/compiler/xla/
shape_layout.cc 25 Status ShapeLayout::CopyLayoutFromShape(const Shape& other_shape) {
26 if (!ShapeUtil::Compatible(other_shape, shape_)) {
28 ShapeUtil::HumanString(other_shape),
31 shape_ = other_shape;
shape_layout.h 50 // Copies the layout from the given shape into this ShapeLayout. 'other_shape'
52 Status CopyLayoutFromShape(const Shape& other_shape);
layout_util_test.cc 49 Shape other_shape = local
66 Shape other_tuple2 = ShapeUtil::MakeTupleShape({shape, other_shape});
  /external/tensorflow/tensorflow/compiler/xla/service/cpu/
cpu_layout_assignment_test.cc 220 Shape other_shape = ShapeUtil::MakeShapeWithLayout(F32, {100, 24}, {0, 1}); local
222 auto constant_shape = ShapeUtil::MakeTupleShape({other_shape, rhs_shape});

Completed in 114 milliseconds