Home
last modified time | relevance | path

Searched refs:parameter_map (Results 1 – 12 of 12) sorted by relevance

/external/tensorflow/tensorflow/core/grappler/optimizers/data/
Dauto_shard.cc259 if ((config->parameter_map().find("num_workers") == in Init()
260 config->parameter_map().end())) { in Init()
264 if ((config->parameter_map().find("index") == in Init()
265 config->parameter_map().end())) { in Init()
269 num_workers_ = config->parameter_map().at("num_workers").i(); in Init()
270 index_ = config->parameter_map().at("index").i(); in Init()
Dmeta_optimizer.cc88 auto& optimizers = config->parameter_map().at("optimizers").list().s(); in Init()
Drebatch.cc37 num_workers_ = config->parameter_map().at("num_workers").i(); in Init()
/external/tensorflow/tensorflow/python/compiler/tensorrt/
Dtrt_convert_test.py88 self.assertTrue(key in trt_optimizer.parameter_map)
89 self.assertEqual(10, trt_optimizer.parameter_map["minimum_segment_size"].i)
90 self.assertEqual(128, trt_optimizer.parameter_map["max_batch_size"].i)
91 self.assertEqual(True, trt_optimizer.parameter_map["is_dynamic_op"].b)
93 trt_optimizer.parameter_map["max_workspace_size_bytes"].i)
96 trt_optimizer.parameter_map["precision_mode"].s)
97 self.assertEqual(2, trt_optimizer.parameter_map["maximum_cached_engines"].i)
99 [1, 128], trt_optimizer.parameter_map["cached_engine_batches"].list.i)
Dtrt_convert.py543 optimizer.parameter_map["minimum_segment_size"].i = minimum_segment_size
544 optimizer.parameter_map["max_batch_size"].i = max_batch_size
545 optimizer.parameter_map["is_dynamic_op"].b = is_dynamic_op
546 optimizer.parameter_map[
548 optimizer.parameter_map["precision_mode"].s = _to_bytes(precision_mode)
549 optimizer.parameter_map["maximum_cached_engines"].i = maximum_cached_engines
551 optimizer.parameter_map["cached_engine_batches"].list.i.extend(
553 optimizer.parameter_map["use_calibration"].b = use_calibration
554 optimizer.parameter_map["use_function_backup"].b = use_function_backup
/external/v8/src/builtins/
Dbuiltins-arguments-gen.cc105 Node* parameter_map = nullptr; in AllocateArgumentsObject() local
109 parameter_map = InnerAllocate(arguments, parameter_map_offset); in AllocateArgumentsObject()
111 parameter_map); in AllocateArgumentsObject()
114 StoreMapNoWriteBarrier(parameter_map, sloppy_elements_map); in AllocateArgumentsObject()
116 StoreObjectFieldNoWriteBarrier(parameter_map, FixedArray::kLengthOffset, in AllocateArgumentsObject()
127 return std::tuple<Node*, Node*, Node*>(result, arguments, parameter_map); in AllocateArgumentsObject()
/external/v8/src/runtime/
Druntime-scopes.cc407 Handle<FixedArray> parameter_map = in NewSloppyArguments() local
409 parameter_map->set_map( in NewSloppyArguments()
412 result->set_elements(*parameter_map); in NewSloppyArguments()
419 parameter_map->set(0, *context); in NewSloppyArguments()
420 parameter_map->set(1, *arguments); in NewSloppyArguments()
437 parameter_map->set_the_hole(i + 2); in NewSloppyArguments()
448 parameter_map->set(parameter + 2, slot); in NewSloppyArguments()
/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dtrt_optimization_pass.cc45 const auto params = config->parameter_map(); in Init()
/external/v8/src/
Dlookup.cc777 FixedArray* parameter_map = FixedArray::cast(receiver->elements()); in TransitionToAccessorPair() local
778 uint32_t length = parameter_map->length() - 2; in TransitionToAccessorPair()
780 parameter_map->set(number_ + 2, ReadOnlyRoots(heap()).the_hole_value()); in TransitionToAccessorPair()
Delements.cc3712 Handle<FixedArrayBase> parameter_map) { in SetLengthImpl() argument
Dobjects.cc9144 FixedArray* parameter_map = FixedArray::cast(current->elements()); in DictionaryElementsInPrototypeChainOnly() local
9145 Object* arguments = parameter_map->get(1); in DictionaryElementsInPrototypeChainOnly()
/external/tensorflow/tensorflow/core/protobuf/
Drewriter_config.proto168 map<string, AttrValue> parameter_map = 2; field