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

  /external/tensorflow/tensorflow/compiler/xla/service/gpu/
instruction_fusion.cc 91 int64 num_output_buffers = ShapeUtil::SubshapeCount(a->shape()) +
92 ShapeUtil::SubshapeCount(b->shape());
multi_output_fusion_test.cc 618 EXPECT_LE(instr->operand_count() + ShapeUtil::SubshapeCount(instr->shape()),
  /external/tensorflow/tensorflow/compiler/xla/
shape_util.h 476 static int64 SubshapeCount(const Shape& shape);
    [all...]
shape_util.cc 351 /* static */ int64 ShapeUtil::SubshapeCount(const Shape& shape) {
    [all...]

Completed in 87 milliseconds