Searched defs:GetTensorShape (Results 1 – 7 of 7) sorted by relevance
34 inline RuntimeShape GetTensorShape(const TfLiteTensor* tensor) { in GetTensorShape() function
26 inline RuntimeShape GetTensorShape(std::vector<int32_t> data) { in GetTensorShape() function
30 const RuntimeShape GetTensorShape(const TfLiteEvalTensor* tensor) { in GetTensorShape() function
436 struct GetTensorShape : public OpRewritePattern<GetShapeOp> { struct439 LogicalResult matchAndRewrite(GetShapeOp shape_op, in matchAndRewrite()
42 TensorShape GetTensorShape(OpKernelContext* ctx, size_t src_index) { in GetTensorShape() function in tensorflow::MklAddNOp
80 std::vector<tensorflow::int64> GetTensorShape(const tensorflow::Tensor& t) { in GetTensorShape() function
595 std::vector<int> GetTensorShape(int index) { in GetTensorShape() function