Searched defs:GetTensorShape (Results 1 – 5 of 5) sorted by relevance
34 inline RuntimeShape GetTensorShape(const TfLiteTensor* tensor) { in GetTensorShape() function
28 inline RuntimeShape GetTensorShape(std::vector<int32_t> data) { in GetTensorShape() function
41 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
395 std::vector<int> GetTensorShape(int index) { in GetTensorShape() function