Searched defs:SetShape (Results 1 – 8 of 8) sorted by relevance
55 void SetShape(std::initializer_list<T> data) { in SetShape() function in tflite::__anonb54fb7560111::ScatterNdOpModel
61 void SetShape(std::initializer_list<ShapeType> data) { in SetShape() function in tflite::__anonae4de6640111::BroadcastToOpModel
72 void SetShape(TfLiteTensor* tensor, std::initializer_list<int> dims) { in SetShape() function in tflite::__anona9a8652d0111::KernelUtilTest
176 Status RemoteTensorHandleData::SetShape(const TensorShape& shape) { in SetShape() function in tensorflow::RemoteTensorHandleData
51 void FlexModelTest::SetShape(int tensor_index, const std::vector<int>& values) { in SetShape() function in tflite::flex::testing::FlexModelTest
257 Status ShapeRefiner::SetShape(const Node* node, int output_port, in SetShape() function in tensorflow::ShapeRefiner
169 TEST_F(ShapeRefinerTest, SetShape) { in TEST_F() argument
362 TEST(CAPI, SetShape) { in TEST() argument