Searched refs:tensorIndices (Results 1 – 1 of 1) sorted by relevance
320 convertTensorValues(ValueRange values, const DenseSet<int64_t> &tensorIndices, in convertTensorValues() argument326 if (tensorIndices.contains(idx)) { in convertTensorValues()343 const DenseSet<int64_t> &tensorIndices, in getYieldedValues() argument346 values, tensorIndices, [&](Value val, int64_t index) { in getYieldedValues()358 const DenseSet<int64_t> &tensorIndices) { in getBbArgReplacements() argument363 if (tensorIndices.contains(idx)) { in getBbArgReplacements()