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 178 output_rank = std::max(output_rank, ShapeUtil::TrueRank(subshape));
191 return ShapeUtil::TrueRank(operand->shape()) >= output_rank;
layout_assignment.cc     [all...]
  /external/tensorflow/tensorflow/compiler/xla/
shape_util.h 303 static int64 TrueRank(const Shape& shape);
315 return shape.IsArray() && TrueRank(shape) == 0;
    [all...]
shape_util.cc 137 /* static */ int64 ShapeUtil::TrueRank(const Shape& shape) {
    [all...]

Completed in 87 milliseconds