Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/
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 …]
Dshape_tree.h125 T* mutable_element(ShapeIndexView index) { return &find(index)->second; } in mutable_element() function
/external/tensorflow/tensorflow/compiler/xla/service/
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()
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()
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()
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()
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()
Dhlo_input_output_alias_config.cc46 (*alias_.mutable_element(output_index)) = in SetUpAlias()
223 *(param_has_seen[alias.parameter_number].mutable_element( in Verify()
Dexecutable.h101 *buffers_.mutable_element(index) = std::move(buffer); in SetBuffer()
122 return buffers_.mutable_element(index); in MutableBuffer()
Dexecutable.cc37 auto buffer = buffers_.mutable_element(index)->Release(); in ~ExecutionInput()
57 *buffers_.mutable_element(index) = std::move(buffer); in SetUnownedBuffer()
Dhlo_runner.cc146 *buffer_tree.mutable_element(index) = se::OwningDeviceMemory{ in ExecutionInputsFromScopedShapedBuffers()
150 *buffer_tree.mutable_element(index) = execution_input_buffer; in ExecutionInputsFromScopedShapedBuffers()
Dhlo_liveness_analysis.cc74 bool& alive = *liveness->mutable_element(shape_index); in MarkLiveAtIndex()
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()
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()
Dhlo_computation_test.cc350 *indices_to_copy.mutable_element({0}) = true; in TEST_F()
Dhlo_sharding.cc501 *result.mutable_element(ShapeIndex({})) = *it; in AsShapeTree()
Dhlo_computation.cc853 *copies_added->mutable_element(leaf_index) = copy; in DeepCopyInstruction()
Dhlo_dataflow_analysis.cc544 return *GetInstructionValueSet(instruction).mutable_element(index); in GetValueSet()
/external/tensorflow/tensorflow/core/tpu/kernels/
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()
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()
/external/libprotobuf-mutator/examples/xml/
Dxml_converter.cc108 message.mutable_document()->mutable_element()->add_content()->set_char_data( in main()
/external/tensorflow/tensorflow/compiler/xrt/
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()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dhlo_to_ir_bindings.cc167 *(base_ptrs_[&hlo].mutable_element(shape_index)) = typed_ir_value; in BindHloToIrValue()