| /external/tensorflow/tensorflow/lite/delegates/flex/ |
| D | delegate_test.cc | 64 SetShape(0, {2, 2, 1}); in TEST_F()
66 SetShape(3, {2, 2, 1}); in TEST_F()
83 SetShape(0, {2, 2}); in TEST_F()
85 SetShape(1, {2, 2}); in TEST_F()
102 SetShape(0, {2, 2}); in TEST_F()
104 SetShape(1, {2, 2}); in TEST_F()
125 SetShape(0, {2, 2, 1}); in TEST_F()
127 SetShape(3, {2, 2, 1}); in TEST_F()
150 SetShape(0, {2, 2, 2, 1}); in TEST_F()
166 SetShape(0, {2, 2, 1}); in TEST_F()
[all …]
|
| D | kernel_test.cc | 117 SetShape(0, {2, 2, 1}); in TEST_F()
119 SetShape(3, {2, 2, 1}); in TEST_F()
128 SetShape(0, {2, 3, 1}); in TEST_F()
130 SetShape(3, {2, 3, 1}); in TEST_F()
160 SetShape(0, {2, 2, 1}); in TEST_F()
178 SetShape(0, {2, 2, 1}); in TEST_F()
214 SetShape(0, {2, 2, 1}); in TEST_F()
216 SetShape(3, {2, 2, 1}); in TEST_F()
263 SetShape(0, {2, 2, 2, 1}); in TEST_F()
265 SetShape(1, {2, 2, 1}); in TEST_F()
[all …]
|
| D | test_util.h | 75 void SetShape(int tensor_index, const std::vector<int>& values);
|
| 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/lite/kernels/ |
| D | kernel_util_test.cc | 43 void SetShape(TfLiteTensor* tensor, std::initializer_list<int> dims) { in SetShape() function in tflite::__anon5b8f934f0111::KernelUtilTest
70 SetShape(&tensor1_, {1, 2, 3}); in TEST_F()
73 SetShape(&tensor2_, {1, 2}); in TEST_F()
76 SetShape(&tensor2_, {1, 2, 3, 4}); in TEST_F()
79 SetShape(&tensor2_, {1, 2, 3}); in TEST_F()
82 SetShape(&tensor2_, {}); in TEST_F()
85 SetShape(&tensor1_, {}); in TEST_F()
91 SetShape(&tensor1_, {1, 2}); in TEST_F()
92 SetShape(&tensor2_, {1, 3}); in TEST_F()
100 SetShape(&tensor1_, {1, 1}); in TEST_F()
[all …]
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | shape_refiner_test.cc | 184 TEST_F(ShapeRefinerTest, SetShape) { in TEST_F() argument
195 TF_ASSERT_OK(m.SetShape(a.node(), 0, h)); in TEST_F()
200 TF_ASSERT_OK(m.SetShape(a.node(), 0, h2)); in TEST_F()
204 ASSERT_FALSE(m.SetShape(a.node(), 1, h).ok()); in TEST_F()
205 ASSERT_FALSE(m.SetShape(a.node(), -1, h).ok()); in TEST_F()
209 ASSERT_FALSE(m.SetShape(b.node(), 0, h).ok()); in TEST_F()
213 ASSERT_FALSE(m.SetShape(a.node(), 0, h).ok()); in TEST_F()
|
| D | shape_refiner.h | 115 Status SetShape(const Node* node, int output_port,
|
| D | shape_refiner.cc | 279 Status ShapeRefiner::SetShape(const Node* node, int output_port, in SetShape() function in tensorflow::ShapeRefiner
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | shape_inference.cc | 90 TF_RETURN_IF_ERROR(shape_refiner->SetShape(n, 0, handle)); in PropagateShapes()
|
| /external/tensorflow/tensorflow/core/graph/ |
| D | graph_constructor.cc | 623 s = refiner_->SetShape(node, i, h); in ValidateShape()
|
| /external/tensorflow/tensorflow/core/kernels/hexagon/ |
| D | graph_transferer.cc | 365 TF_RETURN_IF_ERROR(shape_refiner->SetShape(input_node, i, handle)); in TransformGraphToAddAggregatedInputNode()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | remote_fused_graph_execute_utils.cc | 464 status = shape_refiner->SetShape(node, 0, handle); in PropagateShapeInference()
|
| /external/tensorflow/tensorflow/c/ |
| D | c_api_test.cc | 359 TEST(CAPI, SetShape) { in TEST() argument
|
| D | c_api.cc | 1125 status->status = graph->refiner.SetShape(node, output.index, new_shape); in TF_GraphSetTensorShape()
|