Home
last modified time | relevance | path

Searched defs:tuple_shapes (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/python/tpu/
Dtpu_feed.py136 tuple_shapes=None, argument
256 def tuple_shapes(self): member in InfeedQueue
260 def set_tuple_shapes(self, tuple_shapes): argument
755 tuple_shapes=None, argument
/external/tensorflow/tensorflow/compiler/xla/python_api/
Dxla_shape.py82 def tuple_shapes(self): member in Shape
/external/tensorflow/tensorflow/compiler/xla/
Dshape.h118 const Shape& tuple_shapes(int index) const { return tuple_shapes_.at(index); } in tuple_shapes() function
125 const std::vector<Shape>& tuple_shapes() const { return tuple_shapes_; } in tuple_shapes() function
Dxla_data.proto213 repeated ShapeProto tuple_shapes = 4; field
/external/tensorflow/tensorflow/compiler/xla/service/
Dtriangular_solve_expander.cc179 std::vector<Shape> tuple_shapes = { in InvertDiagonalBlocks() local