Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/lite/kernels/perception/
Dmax_unpooling_2d_test.cc102 std::vector<int32_t> indices_data{0, 3, 4, 7, 8, 11, 12, 15}; in TEST() local
113 model.SetIndices(indices_data); in TEST()
127 std::vector<int32_t> indices_data{0, 5, 10, 13, 19, 20, 27, 31}; in TEST() local
138 model.SetIndices(indices_data); in TEST()
154 std::vector<int32_t> indices_data{6, 9, 16, 19}; in TEST() local
165 model.SetIndices(indices_data); in TEST()
182 std::vector<int32_t> indices_data{2, 23, 8, 9, 12, 15, 40, 43, 44, 47, 72, in TEST() local
195 model.SetIndices(indices_data); in TEST()
223 std::vector<int32_t> indices_data{2, 23, 8, 9, 12, 15, 40, 43, 44, 47, 72, in TEST() local
236 model.SetIndices(indices_data); in TEST()
Dmax_unpooling_2d.cc34 const float* input_data, const int32_t* indices_data, in MaxUnpooling() argument
47 int idx = indices_data[input_offset]; in MaxUnpooling()
Dmax_pool_with_argmax.cc35 T* output_data, int32_t* indices_data) { in MaxPool() argument
86 indices_data[output_idx] = in MaxPool()
/external/tensorflow/tensorflow/lite/kernels/
Dembedding_lookup_sparse_test.cc57 std::initializer_list<int> indices_data, in SetInput() argument
61 PopulateTensor(indices_, indices_data); in SetInput()
Dsparse_to_dense.cc93 const auto indices_data = GetTensorData<T>(indices); in GetIndicesVector() local
95 std::vector<T> index({0, 0, 0, indices_data[i]}); in GetIndicesVector()
/external/tensorflow/tensorflow/lite/experimental/kernels/
Dctc_beam_search_decoder.cc156 int32_t* indices_data = GetTensorData<int32_t>(indices); in StoreAllDecodedSequences() local
166 indices_data[offset * 2] = b; in StoreAllDecodedSequences()
167 indices_data[offset * 2 + 1] = t; in StoreAllDecodedSequences()
/external/tensorflow/tensorflow/lite/kernels/internal/reference/
Dreference_ops.h1097 const IndicesT* indices_data, in GatherNd() argument
1105 from_pos += indices_data[i * res.indices_nd + j] * res.dims_to_count[j]; in GatherNd()
1117 const IndicesT* indices_data, in GatherNdString() argument
1127 from_pos += indices_data[i * res.indices_nd + j] * res.dims_to_count[j]; in GatherNdString()
1139 const IndicesT* indices_data, in ScatterNd() argument
1169 IndicesT idx = indices_data[i * indices_nd + j]; in ScatterNd()
/external/deqp/external/openglcts/modules/gl/
Dgl4cVertexAttrib64BitTest.cpp2673 std::vector<GLuint> indices_data; in testInit() local
2674 indices_data.resize(m_n_vertices); in testInit()
2677 indices_data[i] = i; in testInit()
2697 …gl.bufferData(GL_ELEMENT_ARRAY_BUFFER, m_n_vertices * sizeof(GLuint), &indices_data[0], GL_STATIC_… in testInit()
Dgl4cShadingLanguage420PackTests.cpp16549 static const GLuint indices_data[4] = { 2, 1, 0, 3 }; in prepareUniforms() local
16561 program.uniform("indices", Utils::UINT, 1 /* n_cols */, 4 /* n_rows */, indices_data); in prepareUniforms()