Searched defs:c_shape (Results 1 – 10 of 10) sorted by relevance
212 void ToC(const xla::Shape& xla_shape, XLA_Shape* c_shape) { in ToC()233 xla::Shape FromC(const XLA_Shape* c_shape) { in FromC()252 void Free(XLA_Shape* c_shape) { in Free()318 XLA_ShapeIndex c_shape; in ToC() local327 xla::ShapeIndex FromC(XLA_ShapeIndex* c_shape) { in FromC()
134 XLA_Shape c_shape; in TransferLiteralFromOutfeed() local214 XLA_Shape c_shape; in GetByteSizeRequirement() local280 XLA_Shape c_shape; in WriteSingleTupleIndexTable() local
119 XLA_Shape c_shape; in ShapeSize() local
56 Shape c_shape = ShapeUtil::MakeShape(S64, {}); in TEST() local
35 XLA_Shape c_shape; in GetTPUInfeedLayout() local
119 XLA_Shape c_shape; in ShapeSizeCompact() local128 XLA_Shape c_shape; in ShapeSizeCompactRaw() local
195 XLA_Shape c_shape; in ShapeSizeBytesFunction() local
58 XLA_Shape c_shape; in GetTPUInfeedLayout() local
527 TensorShape c_shape; in Compute() local
3371 const PartialTensorShape c_shape( in ConstantPushDown() local