| /external/tensorflow/tensorflow/compiler/xla/ |
| D | shape_tree_test.cc | 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 …]
|
| D | shape_tree.h | 122 T* mutable_element(ShapeIndexView index);
270 *result.mutable_element(index) = func(t); in Map()
279 *result.mutable_element(index) = func(t); in Map()
602 T* ShapeTree<T>::mutable_element(ShapeIndexView index) { in mutable_element() function
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | tuple_points_to_analysis.cc | 111 mutable_element(index)->push_back(&buffer); in AddPointedToBuffer()
121 tree_.mutable_element(index)->tuple_sources.insert(tuple); in add_tuple_source()
259 points_to_set.mutable_element(/*index=*/{})->clear(); in HandleCopy()
355 *points_to_set.mutable_element(target_index) = points_to; in HandleCopyDone()
372 auto top_buffer = points_to_set.mutable_element(ShapeIndex({})); in HandleSend()
377 auto context_buffer = points_to_set.mutable_element(ShapeIndex({1})); in HandleSend()
381 auto token_buffer = points_to_set.mutable_element(ShapeIndex({2})); in HandleSend()
395 *points_to_set.mutable_element(target_index) = points_to; in HandleSend()
431 *points_to_set.mutable_element(target_index) = points_to; in HandleTuple()
471 points_to_set.mutable_element({})->clear(); in HandleTupleSelect()
|
| D | shaped_buffer.h | 86 *buffers_.mutable_element(index) = buffer; in set_buffer()
186 *buffers_.mutable_element(index) = buffer.Release(); in set_buffer()
188 *buffers_.mutable_element(index) = se::DeviceMemoryBase(); in set_buffer()
|
| D | hlo_replication_analysis.cc | 315 *shape_tree.mutable_element(index) = in ComputeHloReplicationOnComputation()
343 *shape_tree.mutable_element(index) = in ComputeHloReplication()
356 *shape_tree.mutable_element(index) = true; in ComputeHloReplication()
|
| D | tuple_points_to_analysis.h | 119 BufferList* mutable_element(const ShapeIndex& index) { in mutable_element() function
120 return &tree_.mutable_element(index)->buffers; in mutable_element()
|
| D | hlo_sharding_test.cc | 245 *shape_tree1.mutable_element({0}) = HloSharding::Replicate(); in TEST_F()
249 *shape_tree2.mutable_element({0}) = HloSharding::AssignDevice(0); in TEST_F()
259 *shape_tree1.mutable_element({0}) = HloSharding::AssignDevice(0); in TEST_F()
263 *shape_tree2.mutable_element({0}) = HloSharding::AssignDevice(0); in TEST_F()
|
| D | hlo_input_output_alias_config.cc | 46 (*alias_.mutable_element(output_index)) = in SetUpAlias()
223 *(param_has_seen[alias.parameter_number].mutable_element( in Verify()
|
| D | hlo_liveness_analysis.cc | 79 *it->second.mutable_element(shape_index) = true; in MarkLiveAtIndex()
102 *it->second.mutable_element(shape_index) = true; in MarkLiveAtAllIndices()
|
| D | executable.h | 103 *buffers_.mutable_element(index) = std::move(buffer); in SetBuffer()
124 return buffers_.mutable_element(index); in MutableBuffer()
|
| D | executable.cc | 38 auto buffer = buffers_.mutable_element(index)->Release(); in ~ExecutionInput()
58 *buffers_.mutable_element(index) = std::move(buffer); in SetUnownedBuffer()
|
| D | hlo_runner.cc | 138 *buffer_tree.mutable_element(index) = se::OwningDeviceMemory{ in ExecutionInputsFromScopedShapedBuffers()
142 *buffer_tree.mutable_element(index) = execution_input_buffer; in ExecutionInputsFromScopedShapedBuffers()
|
| D | copy_insertion.cc | 369 *(param_indices_to_copy.mutable_element(alias.parameter_index)) = in AddCopiesForAliasedInputOutputs()
371 *(output_indices_to_copy.mutable_element(output_index)) = true; in AddCopiesForAliasedInputOutputs()
1064 *it->second.mutable_element(index) = true; in AddSpecialCaseCopies()
|
| D | hlo_sharding.cc | 336 *result.mutable_element(ShapeIndex({})) = *it; in AsShapeTree()
|
| D | dynamic_dimension_inference.cc | 1350 dynamic_output_mapping.mutable_element(index)->emplace( in HandleConditional()
1445 dynamic_output_mapping.mutable_element(index)->emplace(dim, in HandleWhile()
|
| D | hlo_computation_test.cc | 368 *indices_to_copy.mutable_element({0}) = true; in TEST_F()
|
| D | hlo_computation.cc | 810 *copies_added->mutable_element(leaf_index) = copy; in DeepCopyInstruction()
|
| D | hlo_dataflow_analysis.cc | 375 return *GetInstructionValueSet(instruction).mutable_element(index); in GetValueSet()
|
| /external/libprotobuf-mutator/examples/xml/ |
| D | xml_converter.cc | 108 message.mutable_document()->mutable_element()->add_content()->set_char_data( in main()
|
| /external/tensorflow/tensorflow/compiler/xrt/ |
| D | xrt_state.cc | 396 *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/core/tpu/kernels/ |
| D | tpu_execute_op.cc | 287 TF_ASSIGN_OR_RETURN(*input_buffers->buffers.mutable_element({}), in BuildComputationInputs()
301 auto* in_buffer = input_buffers->buffers.mutable_element(in_index); in BuildComputationInputs()
476 *output_buffers->owned_buffers.mutable_element(out_index) = false; in AllocateOutputTensors()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | hlo_to_ir_bindings.cc | 166 *(base_ptrs_[&hlo].mutable_element(shape_index)) = typed_ir_value; in BindHloToIrValue()
|