Home
last modified time | relevance | path

Searched refs:subshape_ (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/
Dliteral.h360 const Shape& subshape() const { return *subshape_; } in subshape()
361 void set_subshape(const Shape* subshape) { subshape_ = subshape; } in set_subshape()
519 const Shape* subshape_ = nullptr; variable
Dliteral.cc388 CHECK(subshape_ != nullptr); in CopyFrom()
389 CHECK(src.subshape_ != nullptr); in CopyFrom()
/external/tensorflow/tensorflow/compiler/xla/service/
Dpattern_matcher.h902 : index_(index), subshape_(subshape) {}
915 subshape_.DescribeTo(os, indent + kIndentInc);
932 if (!subshape_.Match(GetSubshape(shape), option)) {
940 ShapePattern<SubshapeType, SubshapeImpl> subshape_;