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.h170 Index inputIndices[] = {0, 0};
176 inputIndices[0] += idx0 * m_inputStrides[i];
177 inputIndices[1] += idx1 * m_inputStrides[i];
181 inputIndices[0] += indices[0] * m_inputStrides[0];
182 inputIndices[1] += indices[1] * m_inputStrides[0];
187 inputIndices[0] += idx0 * m_inputStrides[i];
188 inputIndices[1] += idx1 * m_inputStrides[i];
192 inputIndices[0] += indices[0] * m_inputStrides[NumDims-1];
193 inputIndices[1] += indices[1] * m_inputStrides[NumDims-1];
195 if (inputIndices[1] - inputIndices[0] == PacketSize - 1) {
[all …]
DTensorPatch.h197 Index inputIndices[2] = {0, 0};
210 inputIndices[0] += (patchIdx[0] + offsetIdx[0]) * m_inputStrides[i];
211 inputIndices[1] += (patchIdx[1] + offsetIdx[1]) * m_inputStrides[i];
225 inputIndices[0] += (patchIdx[0] + offsetIdx[0]) * m_inputStrides[i];
226 inputIndices[1] += (patchIdx[1] + offsetIdx[1]) * m_inputStrides[i];
229 inputIndices[0] += (patchIndices[0] + patchOffsets[0]);
230 inputIndices[1] += (patchIndices[1] + patchOffsets[1]);
232 if (inputIndices[1] - inputIndices[0] == PacketSize - 1) {
233 PacketReturnType rslt = m_impl.template packet<Unaligned>(inputIndices[0]);
238 values[0] = m_impl.coeff(inputIndices[0]);
[all …]
DTensorMorphing.h415 Index inputIndices[] = {0, 0};
421 inputIndices[0] += (idx0 + m_offsets[i]) * m_inputStrides[i];
422 inputIndices[1] += (idx1 + m_offsets[i]) * m_inputStrides[i];
426 inputIndices[0] += (indices[0] + m_offsets[0]);
427 inputIndices[1] += (indices[1] + m_offsets[0]);
432 inputIndices[0] += (idx0 + m_offsets[i]) * m_inputStrides[i];
433 inputIndices[1] += (idx1 + m_offsets[i]) * m_inputStrides[i];
437 inputIndices[0] += (indices[0] + m_offsets[NumDims-1]);
438 inputIndices[1] += (indices[1] + m_offsets[NumDims-1]);
440 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/llvm-project/mlir/lib/Dialect/Linalg/Transforms/
DLoops.cpp380 static Value getPoolingInput(PoolingOp op, ArrayRef<Value> inputIndices) { in getPoolingInput() argument
385 return getPaddedInput<StdIndexedValue>(op.input(), inputIndices, in getPoolingInput()
390 return input(inputIndices); in getPoolingInput()
/external/deqp/external/vulkancts/modules/vulkan/spirv_assembly/
DvktSpvAsmInstructionTests.cpp11868 const deUint32* const inputIndices = (const deUint32*)&input2Bytes[0]; in compareFP16VectorExtractFunc() local
11877 const deUint32 componentNdx = inputIndices[idx]; in compareFP16VectorExtractFunc()
12122 const deUint32* const inputIndices = (const deUint32*)&input2Bytes[0]; in compareFP16VectorInsertFunc() local
12134 const deUint32 replacedCompNdx = inputIndices[idx]; in compareFP16VectorInsertFunc()