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()
133 XLA_Shape c_shape; in TransferLiteralFromOutfeed() local213 XLA_Shape c_shape; in GetByteSizeRequirement() local279 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
360 XLA_Shape c_shape; in ShapeSizeBytesFunction() local
58 XLA_Shape c_shape; in GetTPUInfeedLayout() local
524 TensorShape c_shape; in Compute() local
3349 const PartialTensorShape c_shape( in ConstantPushDown() local