HomeSort by relevance Sort by last modified time
    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