Home
last modified time | relevance | path

Searched refs:dims_t (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_nodes.cc1588 const auto dims_t = tensor->getDimensions(); in BinaryTensorOpWeight() local
1591 if (dims_t.nbDims != 3) { in BinaryTensorOpWeight()
1606 << "; tensor dims: " << DebugString(dims_t); in BinaryTensorOpWeight()
1608 if (dims_w.nbDims == dims_t.nbDims + 1) { in BinaryTensorOpWeight()
1620 if (dims_w.nbDims == dims_t.nbDims && dims_w.d[0] == dims_t.d[0]) { in BinaryTensorOpWeight()
1624 if (dims_w.d[i] != dims_t.d[i]) { in BinaryTensorOpWeight()
1642 dims_w.d[0] == dims_t.d[dims_t.nbDims - 1]) { in BinaryTensorOpWeight()
1657 std::vector<int> permutation(dims_t.nbDims + 1); in BinaryTensorOpWeight()
1664 permutation[1] = dims_t.nbDims; in BinaryTensorOpWeight()
1665 permutation[dims_t.nbDims] = 1; in BinaryTensorOpWeight()