Searched refs:parameter_map (Results 1 – 12 of 12) sorted by relevance
/external/tensorflow/tensorflow/core/grappler/optimizers/data/ |
D | auto_shard.cc | 259 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()
|
D | meta_optimizer.cc | 88 auto& optimizers = config->parameter_map().at("optimizers").list().s(); in Init()
|
D | rebatch.cc | 37 num_workers_ = config->parameter_map().at("num_workers").i(); in Init()
|
/external/tensorflow/tensorflow/python/compiler/tensorrt/ |
D | trt_convert_test.py | 88 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)
|
D | trt_convert.py | 543 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/ |
D | builtins-arguments-gen.cc | 105 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/ |
D | runtime-scopes.cc | 407 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/ |
D | trt_optimization_pass.cc | 45 const auto params = config->parameter_map(); in Init()
|
/external/v8/src/ |
D | lookup.cc | 777 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()
|
D | elements.cc | 3712 Handle<FixedArrayBase> parameter_map) { in SetLengthImpl() argument
|
D | objects.cc | 9144 FixedArray* parameter_map = FixedArray::cast(current->elements()); in DictionaryElementsInPrototypeChainOnly() local 9145 Object* arguments = parameter_map->get(1); in DictionaryElementsInPrototypeChainOnly()
|
/external/tensorflow/tensorflow/core/protobuf/ |
D | rewriter_config.proto | 168 map<string, AttrValue> parameter_map = 2; field
|