Searched defs:SetShape (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/lite/kernels/ | ||
| D | scatter_nd_test.cc | 53 void SetShape(std::initializer_list<T> data) { in SetShape() function in tflite::__anon97cf11fa0111::ScatterNdOpModel |
| D | kernel_util_test.cc | 47 void SetShape(TfLiteTensor* tensor, std::initializer_list<int> dims) { in SetShape() function in tflite::__anondc1314510111::KernelUtilTest |
| /external/tensorflow/tensorflow/lite/delegates/flex/ | ||
| D | test_util.cc | 51 void FlexModelTest::SetShape(int tensor_index, const std::vector<int>& values) { in SetShape() function in tflite::flex::testing::FlexModelTest |
| /external/tensorflow/tensorflow/core/common_runtime/ | ||
| D | shape_refiner.cc | 252 Status ShapeRefiner::SetShape(const Node* node, int output_port, in SetShape() function in tensorflow::ShapeRefiner |
| D | shape_refiner_test.cc | 169 TEST_F(ShapeRefinerTest, SetShape) { in TEST_F() argument |
| /external/tensorflow/tensorflow/c/ | ||
| D | c_api_test.cc | 359 TEST(CAPI, SetShape) { in TEST() argument |