Searched refs:mutable_element (Results 1 – 19 of 19) sorted by relevance
127 *shape_tree.mutable_element({}) = 42; in TEST_F()129 *shape_tree.mutable_element({}) = 123; in TEST_F()139 *copy.mutable_element({}) = 99; in TEST_F()150 *shape_tree.mutable_element({}) = 1; in TEST_F()151 *shape_tree.mutable_element({0}) = 42; in TEST_F()152 *shape_tree.mutable_element({1}) = 123; in TEST_F()153 *shape_tree.mutable_element({2}) = -100; in TEST_F()196 *shape_tree.mutable_element({0}) = 42; in TEST_F()197 *shape_tree.mutable_element({1, 1}) = 123; in TEST_F()198 *shape_tree.mutable_element({2, 0, 1}) = -100; in TEST_F()[all …]
120 T* mutable_element(ShapeIndexView index);269 *result.mutable_element(index) = func(t); in Map()278 *result.mutable_element(index) = func(t); in Map()560 T* ShapeTree<T>::mutable_element(ShapeIndexView index) { in mutable_element() function
84 *buffers_.mutable_element(index) = buffer; in set_buffer()167 *buffers_.mutable_element(index) = buffer.Forget(); in set_buffer()169 *buffers_.mutable_element(index) = se::DeviceMemoryBase(); in set_buffer()
108 mutable_element(index)->push_back(&buffer); in AddPointedToBuffer()118 tree_.mutable_element(index)->tuple_sources.insert(tuple); in add_tuple_source()256 points_to_set.mutable_element(/*index=*/{})->clear(); in HandleCopy()322 auto top_buffer = points_to_set.mutable_element(ShapeIndex({})); in HandleSend()327 auto context_buffer = points_to_set.mutable_element(ShapeIndex({1})); in HandleSend()331 auto token_buffer = points_to_set.mutable_element(ShapeIndex({2})); in HandleSend()345 *points_to_set.mutable_element(target_index) = points_to; in HandleSend()381 *points_to_set.mutable_element(target_index) = points_to; in HandleTuple()421 points_to_set.mutable_element({})->clear(); in HandleTupleSelect()
210 *shape_tree1.mutable_element({0}) = HloSharding::Replicate(); in TEST_F()214 *shape_tree2.mutable_element({0}) = HloSharding::AssignDevice(0); in TEST_F()224 *shape_tree1.mutable_element({0}) = HloSharding::AssignDevice(0); in TEST_F()228 *shape_tree2.mutable_element({0}) = HloSharding::AssignDevice(0); in TEST_F()
118 BufferList* mutable_element(const ShapeIndex& index) { in mutable_element() function119 return &tree_.mutable_element(index)->buffers; in mutable_element()
46 (*alias_.mutable_element(output_index)) = in SetUpAlias()218 *(param_has_seen[alias.parameter_number].mutable_element( in Verify()
79 *it->second.mutable_element(shape_index) = true; in MarkLiveAtIndex()102 *it->second.mutable_element(shape_index) = true; in MarkLiveAtAllIndices()
357 *(param_indices_to_copy.mutable_element(alias.parameter_index)) = in AddCopiesForAliasedInputOutputs()359 *(output_indices_to_copy.mutable_element(output_index)) = true; in AddCopiesForAliasedInputOutputs()1006 *it->second.mutable_element(index) = true; in AddSpecialCaseCopies()1267 *indices_to_copy.mutable_element(index) = true; in AddCopiesForBufferAssignment()
228 *result.mutable_element(ShapeIndex({})) = *it; in AsShapeTree()
369 *indices_to_copy.mutable_element({0}) = true; in TEST_F()
294 return *GetInstructionValueSet(instruction).mutable_element(index); in GetValueSet()
682 *copies_added->mutable_element(leaf_index) = copy; in DeepCopyInstruction()
329 *parent->buffers_.mutable_element(parent_index) = in MakeSubBuffer()468 *allocation_tmp->buffers_.mutable_element(index) = in MakeTuple()534 *buffers_.mutable_element(dest_index) = source_buffer; in AliasBufferFrom()546 *shaped_tree.mutable_element(buffer.first) = buffer.second->allocation(); in ToDeviceMemoryTree()548 *shaped_tree.mutable_element(buffer.first) = xla::OwningDeviceMemory( in ToDeviceMemoryTree()
64 *buffer_tree.mutable_element(index), in TransferLiteralToInfeed()129 (*shape_tree->mutable_element(*index))->WaitUntilAvailable(); in ShapeTreeToLiteral()
70 infeed_buffers.mutable_element(ShapeIndexView(index, 1)); in ExecuteOnStream()
188 *(base_ptrs_[&hlo].mutable_element(shape_index)) = typed_ir_value; in BindHloToIrValue()
108 message.mutable_document()->mutable_element()->add_content()->set_char_data( in main()
195 shape_tree.mutable_element(original); in DestructureTuple()