Home
last modified time | relevance | path

Searched refs:mutable_element (Results 1 – 25 of 27) sorted by relevance

12

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/
H A Dshape_tree_test.cc135 *shape_tree.mutable_element({}) = 42; in TEST_F()
137 *shape_tree.mutable_element({}) = 123; in TEST_F()
147 *copy.mutable_element({}) = 99; in TEST_F()
158 *shape_tree.mutable_element({}) = 1; in TEST_F()
159 *shape_tree.mutable_element({0}) = 42; in TEST_F()
160 *shape_tree.mutable_element({1}) = 123; in TEST_F()
161 *shape_tree.mutable_element({2}) = -100; in TEST_F()
204 *shape_tree.mutable_element({0}) = 42; in TEST_F()
205 *shape_tree.mutable_element({1, 1}) = 123; in TEST_F()
206 *shape_tree.mutable_element({2, 0, 1}) = -100; in TEST_F()
[all …]
H A Dshape_tree.h125 T* mutable_element(ShapeIndexView index) { return &find(index)->second; } in mutable_element() function
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dtuple_points_to_analysis.cc111 mutable_element(index)->push_back(&buffer); in AddPointedToBuffer()
121 tree_.mutable_element(index)->tuple_sources.insert(tuple); in add_tuple_source()
260 points_to_set.mutable_element(/*index=*/{})->clear(); in HandleCopy()
374 *points_to_set.mutable_element(target_index) = points_to; in HandleAsyncDone()
421 *points_to_set.mutable_element(target_index) = points_to; in HandleCopyDone()
438 auto top_buffer = points_to_set.mutable_element(ShapeIndex({})); in HandleSend()
443 auto context_buffer = points_to_set.mutable_element(ShapeIndex({1})); in HandleSend()
447 auto token_buffer = points_to_set.mutable_element(ShapeIndex({2})); in HandleSend()
461 *points_to_set.mutable_element(target_index) = points_to; in HandleSend()
497 *points_to_set.mutable_element(target_index) = points_to; in HandleTuple()
H A Dshaped_buffer.h85 *buffers_.mutable_element(index) = buffer; in set_buffer()
185 *buffers_.mutable_element(index) = buffer.Release(); in set_buffer()
187 *buffers_.mutable_element(index) = se::DeviceMemoryBase(); in set_buffer()
H A Dhlo_sharding_test.cc252 *shape_tree1.mutable_element({0}) = HloSharding::Replicate(); in TEST_F()
256 *shape_tree2.mutable_element({0}) = HloSharding::AssignDevice(0); in TEST_F()
266 *shape_tree1.mutable_element({0}) = HloSharding::AssignDevice(0); in TEST_F()
270 *shape_tree2.mutable_element({0}) = HloSharding::AssignDevice(0); in TEST_F()
H A Dhlo_replication_analysis.cc383 *shape_tree.mutable_element(index) = in ComputeHloReplicationOnComputation()
417 *shape_tree.mutable_element(index) = in ComputeHloReplication()
428 *shape_tree.mutable_element(index) = in ComputeHloReplication()
434 *shape_tree.mutable_element(index) = in ComputeHloReplication()
H A Dtuple_points_to_analysis.h117 BufferList* mutable_element(const ShapeIndex& index) { in mutable_element() function
118 return &tree_.mutable_element(index)->buffers; in mutable_element()
H A Dhlo_input_output_alias_config.cc46 (*alias_.mutable_element(output_index)) = in SetUpAlias()
223 *(param_has_seen[alias.parameter_number].mutable_element( in Verify()
H A Dexecutable.h101 *buffers_.mutable_element(index) = std::move(buffer); in SetBuffer()
122 return buffers_.mutable_element(index); in MutableBuffer()
H A Dexecutable.cc37 auto buffer = buffers_.mutable_element(index)->Release(); in ~ExecutionInput()
57 *buffers_.mutable_element(index) = std::move(buffer); in SetUnownedBuffer()
H A Dhlo_runner.cc146 *buffer_tree.mutable_element(index) = se::OwningDeviceMemory{ in ExecutionInputsFromScopedShapedBuffers()
150 *buffer_tree.mutable_element(index) = execution_input_buffer; in ExecutionInputsFromScopedShapedBuffers()
H A Ddynamic_dimension_inference.cc1539 dynamic_output_mapping.mutable_element(index)->emplace( in HandleConditional()
1701 dynamic_output_mapping.mutable_element(index)->emplace(dim, in HandleWhile()
1713 if (!dynamic_output_mapping.mutable_element(index)->contains(dim)) { in HandleWhile()
1720 dynamic_output_mapping.mutable_element(index)->emplace( in HandleWhile()
H A Dhlo_liveness_analysis.cc74 bool& alive = *liveness->mutable_element(shape_index); in MarkLiveAtIndex()
H A Dcopy_insertion.cc379 *(param_indices_to_copy.mutable_element(alias.parameter_index)) = in AddCopiesForAliasedInputOutputs()
381 *(output_indices_to_copy.mutable_element(output_index)) = true; in AddCopiesForAliasedInputOutputs()
1876 *it->second.mutable_element(index) = true; in AddSpecialCaseCopies()
H A Dhlo_computation_test.cc350 *indices_to_copy.mutable_element({0}) = true; in TEST_F()
H A Dhlo_sharding.cc501 *result.mutable_element(ShapeIndex({})) = *it; in AsShapeTree()
H A Dhlo_computation.cc853 *copies_added->mutable_element(leaf_index) = copy; in DeepCopyInstruction()
/aosp_15_r20/external/tensorflow/tensorflow/core/tpu/kernels/
H A Dtpu_reshard_variables_op_util.cc158 TF_ASSIGN_OR_RETURN(*input_buffers.mutable_element({}), in BuildInputBuffers()
170 *input_buffers.mutable_element(in_index) = in BuildInputBuffers()
174 *input_buffers.mutable_element(in_index) = *buffer; in BuildInputBuffers()
H A Dtpu_execute_op.cc286 TF_ASSIGN_OR_RETURN(*input_buffers->buffers.mutable_element({}), in BuildComputationInputs()
300 auto* in_buffer = input_buffers->buffers.mutable_element(in_index); in BuildComputationInputs()
475 *output_buffers->owned_buffers.mutable_element(out_index) = false; in AllocateOutputTensors()
/aosp_15_r20/external/private-join-and-compute/private_join_and_compute/
H A Dserver_impl.cc56 *encrypted->mutable_element() = encrypted_element.value(); in EncryptSet()
85 *reencrypted.mutable_element() = reenc.value(); in ComputeIntersection()
H A Dclient_impl.cc60 *element->mutable_element() = encrypted.value(); in ReEncryptSet()
75 *reencrypted.mutable_element() = reenc.value(); in ReEncryptSet()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xrt/
H A Dxrt_state.cc397 *parent->buffers_.mutable_element(parent_index) = nullptr; in MakeSubBuffer()
540 *allocation_tmp->buffers_.mutable_element(index) = in MakeTuple()
633 *buffers_.mutable_element(dest_index) = source_buffer; in AliasBufferFrom()
/aosp_15_r20/external/libprotobuf-mutator/examples/xml/
H A Dxml_converter.cc108 message.mutable_document()->mutable_element()->add_content()->set_char_data( in main()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/
H A Dhlo_to_ir_bindings.cc167 *(base_ptrs_[&hlo].mutable_element(shape_index)) = typed_ir_value; in BindHloToIrValue()
/aosp_15_r20/frameworks/base/tools/aapt2/format/proto/
H A DProtoSerialize.cpp788 pb::XmlElement* pb_element = out_node->mutable_element(); in SerializeXmlToPb()

12