Searched refs:TF_Shape (Results 1 – 8 of 8) sorted by relevance
29 typedef struct TF_Shape TF_Shape; typedef33 TF_CAPI_EXPORT extern TF_Shape* TF_NewShape();36 TF_CAPI_EXPORT extern int TF_ShapeDims(const TF_Shape* shape);41 TF_CAPI_EXPORT extern int64_t TF_ShapeDimSize(const TF_Shape* shape, int d);44 TF_CAPI_EXPORT extern void TF_DeleteShape(TF_Shape* shape);
25 TF_Shape* TF_NewShape() { in TF_NewShape()29 int TF_ShapeDims(const TF_Shape* shape) { in TF_ShapeDims()33 int64_t TF_ShapeDimSize(const TF_Shape* shape, int d) { in TF_ShapeDimSize()37 void TF_DeleteShape(TF_Shape* shape) { delete tensorflow::unwrap(shape); } in TF_DeleteShape()
22 typedef struct TF_Shape TF_Shape; typedef26 DEFINE_CONVERSION_FUNCTIONS(tensorflow::PartialTensorShape, TF_Shape);
68 typedef struct TF_Shape { struct71 } TF_Shape; typedef75 TF_DataType dtype, TF_Shape shape,
137 TF_DataType dtype, TF_Shape shape, in TF_AddFunctionParameter()
39 TF_CAPI_EXPORT extern const TF_Shape* TF_TensorSpecShape(
28 const TF_Shape* TF_TensorSpecShape(const TF_TensorSpec* spec) { in TF_TensorSpecShape()
227 const TF_Shape* shape_a = TF_TensorSpecShape(tensor_spec_a); in TEST_P()236 const TF_Shape* shape_b = TF_TensorSpecShape(tensor_spec_b); in TEST_P()249 const TF_Shape* shape_out = TF_TensorSpecShape(tensor_spec_out); in TEST_P()