Searched defs:SetShape (Results 1 – 8 of 8) sorted by relevance
55 void SetShape(std::initializer_list<T> data) { in SetShape() function in tflite::__anon5656302f0111::ScatterNdOpModel
61 void SetShape(std::initializer_list<ShapeType> data) { in SetShape() function in tflite::__anonaad5f97d0111::BroadcastToOpModel
72 void SetShape(TfLiteTensor* tensor, std::initializer_list<int> dims) { in SetShape() function in tflite::__anon6b7df9260111::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