Searched defs:GetTensorShape (Results 1 – 8 of 8) 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
19 TensorShape GetTensorShape(unsigned int numberOfBatches, in GetTensorShape() function
227 TensorShape GetTensorShape(const ArmComputeType& shapelike, T initial) in GetTensorShape() function
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
594 std::vector<int> GetTensorShape(int index) { in GetTensorShape() function
464 class GetTensorShape : public OpRewritePattern<GetShapeOp> { class