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

  /external/tensorflow/tensorflow/compiler/xla/service/
instruction_fusion.cc 140 output_rank = std::max(output_rank, ShapeUtil::TrueRank(subshape));
152 return ShapeUtil::TrueRank(operand->shape()) >=
layout_assignment.cc     [all...]
  /external/tensorflow/tensorflow/compiler/xla/
shape_util.h 268 static int64 TrueRank(const Shape& shape);
280 return !IsTuple(shape) && !IsOpaque(shape) && TrueRank(shape) == 0;
shape_util.cc 161 /* static */ int64 ShapeUtil::TrueRank(const Shape& shape) {
    [all...]

Completed in 87 milliseconds