OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
Searched
defs:SetShape
(Results
1 - 3
of
3
) sorted by null
/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
...]
/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
...]
/external/tensorflow/tensorflow/core/common_runtime/
shape_refiner.cc
279
Status ShapeRefiner::
SetShape
(const Node* node, int output_port,
[
all
...]
Completed in 122 milliseconds