Searched refs:NameRangeMap (Results 1 – 15 of 15) sorted by relevance
| /external/tensorflow/tensorflow/core/framework/ |
| D | node_def_util_test.cc | 421 NameRangeMap inputs, outputs; in TEST()
425 EXPECT_EQ(NameRangeMap({{"a", {0, 1}}, {"b", {1, 2}}}), inputs); in TEST()
426 EXPECT_EQ(NameRangeMap({{"c", {0, 1}}, {"d", {1, 2}}}), outputs); in TEST()
441 NameRangeMap inputs, outputs; in TEST()
446 EXPECT_EQ(NameRangeMap({{"a", {0, 1}}, {"b", {1, 2}}}), inputs); in TEST()
447 EXPECT_EQ(NameRangeMap({{"c", {0, 1}}}), outputs); in TEST()
455 EXPECT_EQ(NameRangeMap({{"a", {0, 1}}, {"b", {1, 2}}}), inputs); in TEST()
456 EXPECT_EQ(NameRangeMap({{"c", {0, 1}}}), outputs); in TEST()
471 NameRangeMap inputs, outputs; in TEST()
477 EXPECT_EQ(NameRangeMap({{"a", {0, 4}}, {"b", {4, 8}}}), inputs); in TEST()
[all …]
|
| D | memory_types.cc | 32 int GetTotal(const NameRangeMap& name_map) { in GetTotal()
43 void MemoryTypesHelper(const NameRangeMap& name_map, in MemoryTypesHelper()
109 NameRangeMap inp_names; in MemoryTypesForNode()
110 NameRangeMap out_names; in MemoryTypesForNode()
|
| D | node_def_util.h | 296 NameRangeMap; typedef
298 NameRangeMap* inputs, NameRangeMap* outputs);
300 NameRangeMap* inputs, NameRangeMap* outputs);
|
| D | node_def_util.cc | 580 const OpDef& op_def, NameRangeMap* result) { in NameRangesHelper()
594 NameRangeMap* inputs, NameRangeMap* outputs) { in NameRangesForNode()
606 NameRangeMap* inputs, NameRangeMap* outputs) { in NameRangesForNode()
|
| D | shape_inference.h | 774 NameRangeMap input_name_map_;
775 NameRangeMap output_name_map_;
|
| D | graph_to_functiondef.cc | 208 NameRangeMap output_ranges; in GraphToFunctionDef()
|
| D | op_kernel.h | 210 NameRangeMap input_name_map_;
211 NameRangeMap output_name_map_;
|
| D | function.cc | 1642 NameRangeMap output_names; in Define()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/data/ |
| D | map_vectorization.cc | 153 Status GetInputMap(const NodeDef& node, NameRangeMap* result) { in GetInputMap()
161 Status CopyInputs(StringPiece input_name, const NameRangeMap& input_map, in CopyInputs()
176 Status GetInputNodeName(StringPiece input_name, const NameRangeMap& input_map, in GetInputNodeName()
202 NameRangeMap input_map; in AddNewBatchNode()
261 NameRangeMap input_map; in AddNewMapNode()
|
| /external/tensorflow/tensorflow/core/grappler/utils/ |
| D | functions.h | 92 tensorflow::NameRangeMap&& outputs);
122 absl::flat_hash_map<string, tensorflow::NameRangeMap> function_body_outputs_;
|
| D | functions.cc | 42 tensorflow::NameRangeMap outputs_range_map; in RegisterFunctionBodyOutputs()
94 const string& node_name, tensorflow::NameRangeMap&& outputs) { in RegisterFunctionBodyOutputs()
175 const tensorflow::NameRangeMap& outputs = function_body_outputs->second; in ExpandFunctionDefInput()
250 const tensorflow::NameRangeMap& outputs_range_map = is_body_output->second; in AsFunctionDefInput()
|
| /external/tensorflow/tensorflow/python/grappler/ |
| D | item.i | 269 tensorflow::NameRangeMap inputs; in TF_GetColocationGroups()
270 tensorflow::NameRangeMap outputs; in TF_GetColocationGroups()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | xla_op_registry.cc | 400 NameRangeMap input_name_ranges; in CompileTimeConstantInputs()
|
| /external/tensorflow/tensorflow/c/ |
| D | c_api_function.cc | 372 NameRangeMap output_ranges; in GraphToFunctionDef()
|
| D | c_api.cc | 79 using tensorflow::NameRangeMap;
1524 NameRangeMap name_ranges; in TF_OperationOutputListLength()
1546 NameRangeMap name_ranges; in TF_OperationInputListLength()
|