| /external/tensorflow/tensorflow/contrib/lite/kernels/ |
| kernel_util_test.cc | 41 void SetShape(TfLiteTensor* tensor, std::initializer_list<int> dims) {
68 SetShape(&tensor1_, {1, 2, 3});
71 SetShape(&tensor2_, {1, 2});
74 SetShape(&tensor2_, {1, 2, 3, 4});
77 SetShape(&tensor2_, {1, 2, 3});
80 SetShape(&tensor2_, {});
83 SetShape(&tensor1_, {});
89 SetShape(&tensor1_, {1, 2});
90 SetShape(&tensor2_, {1, 3});
98 SetShape(&tensor1_, {1, 1})
[all...] |
| /external/tensorflow/tensorflow/core/common_runtime/ |
| shape_refiner.h | 115 Status SetShape(const Node* node, int output_port,
|
| shape_refiner_test.cc | 150 TEST_F(ShapeRefinerTest, SetShape) {
161 TF_ASSERT_OK(m.SetShape(a.node(), 0, h));
166 TF_ASSERT_OK(m.SetShape(a.node(), 0, h2));
170 ASSERT_FALSE(m.SetShape(a.node(), 1, h).ok());
171 ASSERT_FALSE(m.SetShape(a.node(), -1, h).ok());
175 ASSERT_FALSE(m.SetShape(b.node(), 0, h).ok());
179 ASSERT_FALSE(m.SetShape(a.node(), 0, h).ok());
[all...] |
| shape_refiner.cc | 279 Status ShapeRefiner::SetShape(const Node* node, int output_port,
[all...] |
| /frameworks/ml/nn/common/ |
| OperationsUtils.cpp | 40 bool SetShape(const Shape& in, Shape* out) {
193 return SetShape(in1, out);
220 return SetShape(input, output);
338 return SetShape(input, output);
415 return SetShape(input, output);
[all...] |
| /frameworks/ml/nn/common/include/ |
| OperationsUtils.h | 60 bool SetShape(const Shape& in, Shape* out);
|
| /external/tensorflow/tensorflow/core/graph/ |
| graph_constructor.cc | 586 s = refiner_->SetShape(node, i, h);
[all...] |
| /external/tensorflow/tensorflow/core/kernels/hexagon/ |
| graph_transferer.cc | 358 TF_RETURN_IF_ERROR(shape_refiner->SetShape(input_node, i, handle));
[all...] |
| /external/tensorflow/tensorflow/core/kernels/ |
| remote_fused_graph_execute_utils.cc | 462 status = shape_refiner->SetShape(node, 0, handle);
[all...] |
| /external/tensorflow/tensorflow/c/ |
| c_api.cc | [all...] |
| c_api_test.cc | 338 TEST(CAPI, SetShape) {
[all...] |
