Searched defs:SetShape (Results 1 – 11 of 11) sorted by relevance
49 Status TRT_ShapedWeights::SetShape(DimsAdapter dims) { in SetShape() function in tensorflow::tensorrt::convert::TRT_ShapedWeights
177 Status RemoteTensorHandleData::SetShape(const TensorShape& shape) { in SetShape() function in tensorflow::RemoteTensorHandleData
193 void SetShape(const TensorShape& newShape) { m_Shape = newShape; } in SetShape() function in armnn::TensorInfo
55 void SetShape(std::initializer_list<T> data) { in SetShape() function in tflite::__anon1c86c5740111::ScatterNdOpModel
61 void SetShape(std::initializer_list<ShapeType> data) { in SetShape() function in tflite::__anonbf80f4020111::BroadcastToOpModel
74 void SetShape(TfLiteTensor* tensor, std::initializer_list<int> dims) { in SetShape() function in tflite::__anonf7c1370b0111::KernelUtilTest
51 void FlexModelTest::SetShape(int tensor_index, const std::vector<int>& values) { in SetShape() function in tflite::flex::testing::FlexModelTest
238 void TensorInterface::SetShape(const int64_t* dims, int num_dims) { in SetShape() function in tensorflow::TensorInterface
365 TEST(CAPI, SetShape) { in TEST() argument
273 Status ShapeRefiner::SetShape(const Node* node, int output_port, in SetShape() function in tensorflow::ShapeRefiner
171 TEST_F(ShapeRefinerTest, SetShape) { in TEST_F() argument