Searched refs:update_index (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/ |
| D | dynamic_update_slice_util.cc | 130 auto loop_body_emitter = [&](const IrArray::Index& update_index) -> Status { in EmitDynamicUpdateSliceInPlaceImpl() argument
139 start_multi_index[i], update_index[i]->getType()); in EmitDynamicUpdateSliceInPlaceImpl()
140 output_multi_index[i] = b->CreateAdd(start_index0, update_index[i]); in EmitDynamicUpdateSliceInPlaceImpl()
147 update_array_generator(update_index)); in EmitDynamicUpdateSliceInPlaceImpl()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_evaluator_typed_visitor.h | 2171 absl::Span<const int64> update_index) {
2172 PropagateUpdateIndexScatterDimsToIndexVectorIndex(update_index);
2184 absl::Span<const int64> update_index) {
2186 for (int64 i = 0, e = update_index.size(); i < e; i++) {
2195 index_vector_index_[index_vector_index_i++] = update_index[i];
2294 absl::Span<const int64> update_index) {
2295 PropagateUpdateIndexWindowDimsToInputIndex(update_index);
2309 absl::Span<const int64> update_index) {
2312 input_index_[i] = update_index[input_dim_value_to_update_index_[i]];
2352 std::vector<int64> update_index(updates_shape.dimensions_size());
[all …]
|
| D | elemental_ir_emitter.cc | 2093 llvm_ir::IrArray::Index update_index(update_multi_index, update_hlo->shape(), in EmitElementalDynamicUpdateSlice() local
2096 operand_to_generator.at(update_hlo)(update_index)); in EmitElementalDynamicUpdateSlice()
|
| /external/tensorflow/tensorflow/core/grappler/utils/ |
| D | graph_view.cc | 281 auto& update_index = node->update_index_; in RemoveNode() local
282 if (update_index != internal::kMissingIndex) { in RemoveNode()
284 if (update_index < updated_nodes_size - 1) { in RemoveNode()
286 update_index; in RemoveNode()
287 std::swap(updated_nodes_[update_index], updated_nodes_.back()); in RemoveNode()
290 update_index = internal::kMissingIndex; in RemoveNode()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | resource_op_lifting.cc | 939 int64_t update_index = return_op->getOperand(entry.getFirst()) == in HandleWhileLoop() local
944 entry.getSecond(), update_index}; in HandleWhileLoop()
1044 int64_t update_index = new_output_it == resource_arg_to_new_output.end() in HandleCaseOrIfOp() local
1048 entry.getSecond(), update_index}; in HandleCaseOrIfOp()
|
| /external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/IR/ |
| D | hlo_ops.cc | 3025 llvm::SmallVector<uint64_t, 8> update_index(update_type.getRank(), 0); in fold() local
3035 for (int64_t i = 0; i < update_index.size(); ++i) { in fold()
3037 index_index.push_back(update_index[i]); in fold()
3058 update_index[scatter_dimension_numbers() in fold()
3083 update.getValue<Attribute>(update_index)); in fold()
3088 } while (next_index(update_index, update_type.getShape())); in fold()
|
| /external/vulkan-validation-layers/layers/ |
| D | descriptor_sets.cpp | 919 uint32_t update_index = 0; in PerformWriteUpdate() local
924 for (uint32_t di = 0; di < update_count; ++di, ++update_index) { in PerformWriteUpdate()
925 descriptors_[global_idx + di]->WriteUpdate(update, update_index); in PerformWriteUpdate()
|