Searched defs:ElementsIn (Results 1 - 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/
H A Dshape_util.cc372 /* static */ int64 ShapeUtil::ElementsIn(const Shape& shape) { function in class:ShapeUtil
381 return ElementsIn(shape) == 0;
740 allocated_element_count = ElementsIn(shape);
1120 CHECK_EQ(ElementsIn(input_shape), ElementsIn(output_shape));
1121 if (ElementsIn(input_shape) == 0) {

Completed in 135 milliseconds