Home
last modified time | relevance | path

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

/external/eigen/unsupported/Eigen/CXX11/src/Tensor/
DTensorStriding.h164 Index inputIndices[] = {0, 0};
171 inputIndices[0] += idx0 * m_inputStrides[i];
172 inputIndices[1] += idx1 * m_inputStrides[i];
176 inputIndices[0] += indices[0] * m_inputStrides[0];
177 inputIndices[1] += indices[1] * m_inputStrides[0];
183 inputIndices[0] += idx0 * m_inputStrides[i];
184 inputIndices[1] += idx1 * m_inputStrides[i];
188 inputIndices[0] += indices[0] * m_inputStrides[NumDims-1];
189 inputIndices[1] += indices[1] * m_inputStrides[NumDims-1];
191 if (inputIndices[1] - inputIndices[0] == PacketSize - 1) {
[all …]
DTensorPatch.h208 Index inputIndices[2] = {0, 0};
222 inputIndices[0] += (patchIdx[0] + offsetIdx[0]) * m_inputStrides[i];
223 inputIndices[1] += (patchIdx[1] + offsetIdx[1]) * m_inputStrides[i];
238 inputIndices[0] += (patchIdx[0] + offsetIdx[0]) * m_inputStrides[i];
239 inputIndices[1] += (patchIdx[1] + offsetIdx[1]) * m_inputStrides[i];
242 inputIndices[0] += (patchIndices[0] + patchOffsets[0]);
243 inputIndices[1] += (patchIndices[1] + patchOffsets[1]);
245 if (inputIndices[1] - inputIndices[0] == PacketSize - 1) {
246 PacketReturnType rslt = m_impl.template packet<Unaligned>(inputIndices[0]);
251 values[0] = m_impl.coeff(inputIndices[0]);
[all …]
DTensorMorphing.h559 Index inputIndices[] = {0, 0};
566 inputIndices[0] += (idx0 + m_offsets[i]) * m_inputStrides[i];
567 inputIndices[1] += (idx1 + m_offsets[i]) * m_inputStrides[i];
571 inputIndices[0] += (indices[0] + m_offsets[0]);
572 inputIndices[1] += (indices[1] + m_offsets[0]);
578 inputIndices[0] += (idx0 + m_offsets[i]) * m_inputStrides[i];
579 inputIndices[1] += (idx1 + m_offsets[i]) * m_inputStrides[i];
583 inputIndices[0] += (indices[0] + m_offsets[NumDims-1]);
584 inputIndices[1] += (indices[1] + m_offsets[NumDims-1]);
586 if (inputIndices[1] - inputIndices[0] == packetSize - 1) {
[all …]
/external/tensorflow/tensorflow/java/src/main/java/org/tensorflow/
DGraph.java275 int[] inputIndices, in buildSubgraph() argument
291 inputs[i] = op.output(inputIndices[i]); in buildSubgraph()
326 int[] inputIndices = new int[ninputs]; in whileLoop() local
334 inputIndices[i] = inputs[i].index(); in whileLoop()
338 whileLoop(nativeHandle, inputHandles, inputIndices, name, cgBuilder, bgBuilder); in whileLoop()
464 int[] inputIndices, in addGradients() argument
473 int[] inputIndices, in whileLoop() argument
/external/armnn/src/armnnTfLiteParser/test/
DUnsupported.cpp146 const std::string inputIndices = GetIndicesAsString(0u, numInputs - 1u); in DummyCustomFixture() local
151 "inputs": )" + inputIndices + R"(, in DummyCustomFixture()
156 "inputs": )" + inputIndices + R"(, in DummyCustomFixture()
/external/deqp/external/vulkancts/modules/vulkan/spirv_assembly/
DvktSpvAsmInstructionTests.cpp11967 const deUint32* const inputIndices = (const deUint32*)&input2Bytes[0]; in compareFP16VectorExtractFunc() local
11976 const deUint32 componentNdx = inputIndices[idx]; in compareFP16VectorExtractFunc()
12221 const deUint32* const inputIndices = (const deUint32*)&input2Bytes[0]; in compareFP16VectorInsertFunc() local
12233 const deUint32 replacedCompNdx = inputIndices[idx]; in compareFP16VectorInsertFunc()