Home
last modified time | relevance | path

Searched refs:FindOrNull (Results 1 – 25 of 39) sorted by relevance

12

/external/tensorflow/tensorflow/core/grappler/optimizers/data/
Dmap_fusion.cc61 gtl::FindOrNull(parent_map_node.attr(), "use_inter_op_parallelism"); in MakeFusedNode()
63 gtl::FindOrNull(map_node.attr(), "use_inter_op_parallelism"); in MakeFusedNode()
70 gtl::FindOrNull(parent_map_node.attr(), "preserve_cardinality"); in MakeFusedNode()
72 gtl::FindOrNull(map_node.attr(), "preserve_cardinality"); in MakeFusedNode()
Dfusion_utils.cc131 gtl::FindOrNull(changed_node_names, function_node.name())) { in RenameFunctionNodes()
164 gtl::FindOrNull(changed_input_names, input.name())) { in GetUniqueSignature()
175 gtl::FindOrNull(changed_output_names, output.name())) { in GetUniqueSignature()
198 gtl::FindOrNull(changed_input_names, input)) { in GetUniqueSignature()
Dvectorization_utils.cc256 if (auto found = gtl::FindOrNull(conversion_map_, in AddConversionMapping()
309 auto found = gtl::FindOrNull(conversion_map_, output); in ConvertOutput()
540 if (auto found = gtl::FindOrNull(conversion_map_, tensor)) { in AddUnstackedTensorMappingsHelper()
586 if (auto found = gtl::FindOrNull(conversion_map_, in AddUnstackedTensorMappingsHelper()
Dmeta_optimizer.cc70 const auto* optimizer = gtl::FindOrNull(enabled_optimizers_, name); in ApplyOptimization()
Dmap_vectorization.cc136 auto* shapes_attr = gtl::FindOrNull(node.attr(), "output_shapes"); in IsOutputShapesFullyDefined()
163 const auto* range = gtl::FindOrNull(input_map, input_name); in CopyInputs()
178 const auto* range = gtl::FindOrNull(input_map, input_name); in GetInputNodeName()
Dmap_and_batch_fusion.cc92 if (gtl::FindOrNull(map_node.attr(), key)) { in MakeMapAndBatchNode()
Dmap_and_filter_fusion.cc61 if (gtl::FindOrNull(map_node.attr(), key)) { in MakeFusedNode()
Dhoist_random_uniform.cc62 gtl::FindOrNull(map_node.attr(), "use_inter_op_parallelism")) in MakeStatelessMap()
Dgraph_utils.cc289 if (auto entry = gtl::FindOrNull(name_map, prefix)) { in EnsureNodeNamesUnique()
/external/tensorflow/tensorflow/core/framework/
Dmodel.cc140 if (auto* parameter = gtl::FindOrNull(parameters_, "parallelism")) { in OutputTimeLocked()
225 if (auto* parameter = gtl::FindOrNull(parameters_, "parallelism")) { in OutputTimeLocked()
398 auto node = gtl::FindOrNull(lookup_table_, name); in AddProcessingTime()
468 auto node = gtl::FindOrNull(lookup_table_, name); in RecordElement()
476 auto node = gtl::FindOrNull(lookup_table_, name); in RecordStart()
488 auto node = gtl::FindOrNull(lookup_table_, name); in RecordStop()
500 auto node = gtl::FindOrNull(lookup_table_, name); in RemoveNode()
Dresource_mgr.cc219 if (!gtl::FindOrNull(containers_, container)) { in Cleanup()
/external/tensorflow/tensorflow/core/lib/gtl/
Dmap_util.h35 const typename Collection::value_type::second_type* FindOrNull( in FindOrNull() function
47 typename Collection::value_type::second_type* FindOrNull( in FindOrNull() function
Dmap_util_test.cc36 EXPECT_EQ("bar", *gtl::FindOrNull(m, "foo")); in TEST()
/external/protobuf/src/google/protobuf/compiler/java/
Djava_context.cc173 FindOrNull(field_generator_info_map_, field); in GetFieldGeneratorInfo()
184 FindOrNull(oneof_generator_info_map_, oneof); in GetOneofGeneratorInfo()
/external/tensorflow/tensorflow/core/util/
Dtensor_slice_set.cc67 const TensorSliceSet::SliceInfo* info = gtl::FindOrNull(slices_, str); in Query()
126 const TensorSliceSet::SliceInfo* info = gtl::FindOrNull(slices_, str); in QueryMeta()
/external/tensorflow/tensorflow/core/kernels/
Dload_and_remap_matrix_op.cc225 gtl::FindOrNull(old_row_to_new_row_map, row_index); in Compute()
240 gtl::FindOrNull(old_col_to_new_col_map, old_col); in Compute()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dbuffer_allocations.cc62 tensorflow::gtl::FindOrNull(registered_buffers_, i)) { in Build()
Dthunk_schedule.cc62 if (Thunk** thunk = tensorflow::gtl::FindOrNull(hlo_to_thunk, hlo)) { in ThunkSchedule()
/external/protobuf/src/google/protobuf/util/
Dfield_comparator.cc184 Tolerance* tolerance = FindOrNull(map_tolerance_, &field); in CompareDoubleOrFloat()
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dmeta_optimizer_test.cc674 gtl::FindOrNull(optimization_options, "main"); in TEST_F()
679 gtl::FindOrNull(optimization_options, "MyMul1"); in TEST_F()
684 gtl::FindOrNull(optimization_options, "MyMul2"); in TEST_F()
/external/tensorflow/tensorflow/python/framework/
Dpython_op_gen_internal.cc200 gtl::FindOrNull(inferred_attrs, arg.number_attr()); in ArgTypeName()
220 original_arg = gtl::FindOrNull(inferred_attrs, arg.type_attr()); in ArgTypeName()
251 const string* original_arg = gtl::FindOrNull(inferred_attrs, attr_name); in ArgTypeName()
/external/tensorflow/tensorflow/contrib/tensor_forest/kernels/
Dstats_ops.cc39 using gtl::FindOrNull;
285 if (FindOrNull(locks, id) == nullptr) { in Compute()
/external/tensorflow/tensorflow/core/common_runtime/eager/
Dattr_builder.cc196 auto* t = gtl::FindOrNull(m, attr_name); in AttrTypeByName()
/external/protobuf/src/google/protobuf/stubs/
Dmap_util.h137 FindOrNull(const Collection& collection,
149 FindOrNull(Collection& collection, // NOLINT
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_schedule.cc167 tensorflow::gtl::FindOrNull(new_instruction_uses, instruction); in UpdateComputationSchedule()

12