Home
last modified time | relevance | path

Searched refs:ntuple_shapes (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/stream_executor/tpu/
Dc_api_conversions.cc218 c_shape->ntuple_shapes = xla_shape.tuple_shapes_size(); in ToC()
219 if (c_shape->ntuple_shapes > 0) { in ToC()
220 c_shape->tuple_shapes = new XLA_Shape[c_shape->ntuple_shapes]; in ToC()
221 for (int i = 0; i < c_shape->ntuple_shapes; ++i) { in ToC()
239 tuple_shapes.reserve(c_shape->ntuple_shapes); in FromC()
240 for (int i = 0; i < c_shape->ntuple_shapes; ++i) { in FromC()
259 if (c_shape->ntuple_shapes > 0) { in Free()
260 for (int i = 0; i < c_shape->ntuple_shapes; ++i) { in Free()
Dc_api_decl.h224 int ntuple_shapes; member