Searched defs:update_index (Results 1 – 8 of 8) sorted by relevance
2679 absl::Span<const int64_t> update_index) { in operator ()()2692 absl::Span<const int64_t> update_index) { in PropagateUpdateIndexScatterDimsToIndexVectorIndex()2802 absl::Span<const int64_t> update_index) { in operator ()()2817 absl::Span<const int64_t> update_index) { in PropagateUpdateIndexWindowDimsToInputIndex()2868 std::vector<int64_t> update_index(updates_dims.size()); in HandleScatter() local
2286 llvm_ir::IrArray::Index update_index(update_multi_index, update_hlo->shape(), in EmitElementalDynamicUpdateSlice() local
130 auto loop_body_emitter = [&](const IrArray::Index& update_index) -> Status { in EmitDynamicUpdateSliceInPlaceImpl()
872 int64_t update_index = return_op->getOperand(entry.getFirst()) == in HandleWhileLoop() local977 int64_t update_index = new_output_it == resource_arg_to_new_output.end() in HandleCaseOrIfOp() local
52 fn update_index(table: &mut RawTable<usize>, hash: HashValue, old: usize, new: usize) { in update_index() function
1537 def update_index(m): function
281 auto& update_index = node->update_index_; in RemoveNode() local
919 uint32_t update_index = 0; in PerformWriteUpdate() local