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
266 TensorShape GetTensorShape(const ArmComputeType& shapelike, T initial) in GetTensorShape() function
41 TensorShape GetTensorShape(OpKernelContext* ctx, size_t src_index) { in GetTensorShape() function in tensorflow::MklAddNOp
19 TensorShape GetTensorShape(unsigned int numberOfBatches, in GetTensorShape() function
88 std::vector<int64_t> GetTensorShape(const tensorflow::Tensor& t) { in GetTensorShape() function
639 std::vector<int> GetTensorShape(int index) { in GetTensorShape() function
496 class GetTensorShape : public OpRewritePattern<GetShapeOp> { class