Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/lite/delegates/gpu/common/
Dmodel_builder.cc146 TfLiteIntArray* constant_dims = input1->dims; in ParseInputsWithConstTensor() local
150 constant_dims = input0->dims; in ParseInputsWithConstTensor()
153 if (constant_dims->size <= 0 || NumElements(constant_dims) == 1) { in ParseInputsWithConstTensor()
158 if (CheckIfLinearConvertible(constant_dims).ok()) { in ParseInputsWithConstTensor()
/external/tensorflow/tensorflow/compiler/tests/
Drandomized_tests.cc1935 std::vector<int64> constant_dims = RandomDims(0, 3, 0, 5); in TEST_F() local
1938 std::copy(constant_dims.begin(), constant_dims.end(), in TEST_F()