HomeSort by relevance Sort by last modified time
    Searched refs:mutable_element (Results 1 - 19 of 19) sorted by null

  /external/tensorflow/tensorflow/compiler/xla/
shape_tree_test.cc 127 *shape_tree.mutable_element({}) = 42;
129 *shape_tree.mutable_element({}) = 123;
139 *copy.mutable_element({}) = 99;
150 *shape_tree.mutable_element({}) = 1;
151 *shape_tree.mutable_element({0}) = 42;
152 *shape_tree.mutable_element({1}) = 123;
153 *shape_tree.mutable_element({2}) = -100;
196 *shape_tree.mutable_element({0}) = 42;
197 *shape_tree.mutable_element({1, 1}) = 123;
198 *shape_tree.mutable_element({2, 0, 1}) = -100
    [all...]
shape_tree.h 120 T* mutable_element(ShapeIndexView index);
269 *result.mutable_element(index) = func(t);
278 *result.mutable_element(index) = func(t);
560 T* ShapeTree<T>::mutable_element(ShapeIndexView index) { function in class:xla::ShapeTree
  /external/tensorflow/tensorflow/compiler/xla/service/
shaped_buffer.h 84 *buffers_.mutable_element(index) = buffer;
167 *buffers_.mutable_element(index) = buffer.Forget();
169 *buffers_.mutable_element(index) = se::DeviceMemoryBase();
hlo_sharding_test.cc 210 *shape_tree1.mutable_element({0}) = HloSharding::Replicate();
214 *shape_tree2.mutable_element({0}) = HloSharding::AssignDevice(0);
224 *shape_tree1.mutable_element({0}) = HloSharding::AssignDevice(0);
228 *shape_tree2.mutable_element({0}) = HloSharding::AssignDevice(0);
tuple_points_to_analysis.cc 108 mutable_element(index)->push_back(&buffer);
118 tree_.mutable_element(index)->tuple_sources.insert(tuple);
256 points_to_set.mutable_element(/*index=*/{})->clear();
322 auto top_buffer = points_to_set.mutable_element(ShapeIndex({}));
327 auto context_buffer = points_to_set.mutable_element(ShapeIndex({1}));
331 auto token_buffer = points_to_set.mutable_element(ShapeIndex({2}));
345 *points_to_set.mutable_element(target_index) = points_to;
381 *points_to_set.mutable_element(target_index) = points_to;
421 points_to_set.mutable_element({})->clear();
    [all...]
hlo_input_output_alias_config.cc 46 (*alias_.mutable_element(output_index)) =
218 *(param_has_seen[alias.parameter_number].mutable_element(
tuple_points_to_analysis.h 118 BufferList* mutable_element(const ShapeIndex& index) { function in class:xla::PointsToSet
119 return &tree_.mutable_element(index)->buffers;
hlo_liveness_analysis.cc 79 *it->second.mutable_element(shape_index) = true;
102 *it->second.mutable_element(shape_index) = true;
copy_insertion.cc 357 *(param_indices_to_copy.mutable_element(alias.parameter_index)) =
359 *(output_indices_to_copy.mutable_element(output_index)) = true;
    [all...]
hlo_sharding.cc 228 *result.mutable_element(ShapeIndex({})) = *it;
hlo_computation_test.cc 369 *indices_to_copy.mutable_element({0}) = true;
hlo_dataflow_analysis.cc 294 return *GetInstructionValueSet(instruction).mutable_element(index);
    [all...]
hlo_computation.cc 682 *copies_added->mutable_element(leaf_index) = copy;
    [all...]
  /external/tensorflow/tensorflow/compiler/xla/service/gpu/
gpu_transfer_manager.cc 64 *buffer_tree.mutable_element(index),
129 (*shape_tree->mutable_element(*index))->WaitUntilAvailable();
infeed_thunk.cc 70 infeed_buffers.mutable_element(ShapeIndexView(index, 1));
hlo_to_ir_bindings.cc 188 *(base_ptrs_[&hlo].mutable_element(shape_index)) = typed_ir_value;
  /external/libprotobuf-mutator/examples/xml/
xml_converter.cc 108 message.mutable_document()->mutable_element()->add_content()->set_char_data(
  /external/tensorflow/tensorflow/compiler/xrt/
xrt_state.cc 329 *parent->buffers_.mutable_element(parent_index) =
468 *allocation_tmp->buffers_.mutable_element(index) =
534 *buffers_.mutable_element(dest_index) = source_buffer;
546 *shaped_tree.mutable_element(buffer.first) = buffer.second->allocation();
548 *shaped_tree.mutable_element(buffer.first) = xla::OwningDeviceMemory(
  /external/tensorflow/tensorflow/compiler/xla/python/
local_computation_builder.cc 195 shape_tree.mutable_element(original);

Completed in 295 milliseconds