| /external/tensorflow/tensorflow/core/grappler/optimizers/inference/ |
| D | batch_op_rewriter_test.cc | 52 ::tensorflow::graph_transforms::SetNodeAttr("num_batch_threads", in __anon4fa663a70202()
54 ::tensorflow::graph_transforms::SetNodeAttr("max_batch_size", 16, batch_op); in __anon4fa663a70202()
55 ::tensorflow::graph_transforms::SetNodeAttr("batch_timeout_micros", 10000, in __anon4fa663a70202()
57 ::tensorflow::graph_transforms::SetNodeAttr( in __anon4fa663a70202()
59 ::tensorflow::graph_transforms::SetNodeAttr("max_enqueued_batches", 1000, in __anon4fa663a70202()
63 ::tensorflow::graph_transforms::SetNodeAttr(kEnableAdaptiveSchedulerAttr, in __anon4fa663a70202()
66 ::tensorflow::graph_transforms::SetNodeAttr( in __anon4fa663a70202()
|
| D | batch_op_rewriter.cc | 80 ::tensorflow::graph_transforms::SetNodeAttr(kEnableAdaptiveSchedulerAttr, in SetNodeAttrs()
82 ::tensorflow::graph_transforms::SetNodeAttr( in SetNodeAttrs()
84 ::tensorflow::graph_transforms::SetNodeAttr( in SetNodeAttrs()
86 ::tensorflow::graph_transforms::SetNodeAttr( in SetNodeAttrs()
88 ::tensorflow::graph_transforms::SetNodeAttr( in SetNodeAttrs()
191 ::tensorflow::graph_transforms::SetNodeAttr(kNumBatchThreadsAttr, 0, in Optimize()
202 ::tensorflow::graph_transforms::SetNodeAttr(kNumBatchThreadsAttr, 0, in Optimize()
|
| D | BUILD | 34 "//tensorflow/tools/graph_transforms:transform_utils",
55 "//tensorflow/tools/graph_transforms:transform_utils",
|
| /external/tensorflow/tensorflow/tools/graph_transforms/ |
| D | fold_constants_test.cc | 35 namespace graph_transforms { namespace
166 graph_transforms::TransformFuncContext context; in TestPreserveOutputShapes()
177 graph_transforms::TransformFuncContext context) { in TestConstantFolding()
191 graph_transforms::FoldConstants(graph_def, context, &folded_graph_def)); in TestConstantFolding()
249 TF_ASSERT_OK(graph_transforms::ReplaceSendRecvs( in TestReplaceSendRecvs()
254 graph_transforms::MapNamesToNodes(result_graph_def, &node_map); in TestReplaceSendRecvs()
283 TF_ASSERT_OK(graph_transforms::ReplaceSendRecvs( in TestReplaceSendRecvsPrefixNames()
288 graph_transforms::MapNamesToNodes(result_graph_def, &node_map); in TestReplaceSendRecvsPrefixNames()
320 TF_ASSERT_OK(graph_transforms::RemoveUnusedNodes( in TestRemoveUnusedNodes()
324 graph_transforms::MapNamesToNodes(result_graph_def, &node_map); in TestRemoveUnusedNodes()
[all …]
|
| D | compare_graphs.cc | 36 namespace graph_transforms { namespace
79 return tensorflow::graph_transforms::ParseFlagsAndCompareGraphs(argc, argv); in main()
|
| D | transform_graph_main.cc | 51 return tensorflow::graph_transforms::ParseFlagsAndTransformGraph(argc, argv, in main()
|
| D | file_utils.h | 23 namespace graph_transforms {
|
| D | sort_by_execution_order.cc | 26 namespace graph_transforms { namespace
|
| D | fold_constants_lib.h | 24 namespace graph_transforms {
|
| D | add_default_attributes.cc | 21 namespace graph_transforms { namespace
|
| D | remove_device.cc | 26 namespace graph_transforms { namespace
|
| D | remove_control_dependencies.cc | 20 namespace graph_transforms { namespace
|
| D | transform_graph.h | 26 namespace graph_transforms {
|
| D | README.md | 79 bazel build tensorflow/tools/graph_transforms:transform_graph
80 bazel-bin/tensorflow/tools/graph_transforms/transform_graph \
110 …tps://github.com/tensorflow/tensorflow/blob/master/tensorflow/tools/graph_transforms/summarize_gra…
117 bazel build tensorflow/tools/graph_transforms:summarize_graph
118 bazel-bin/tensorflow/tools/graph_transforms/summarize_graph --in_graph=tensorflow_inception_graph.pb
139 bazel build tensorflow/tools/graph_transforms:transform_graph
140 bazel-bin/tensorflow/tools/graph_transforms/transform_graph \
177 bazel build tensorflow/tools/graph_transforms:transform_graph
178 bazel-bin/tensorflow/tools/graph_transforms/transform_graph \
211 bazel build tensorflow/tools/graph_transforms:transform_graph
[all …]
|
| D | file_utils.cc | 21 namespace graph_transforms { namespace
|
| D | set_device.cc | 20 namespace graph_transforms { namespace
|
| D | rename_op.cc | 26 namespace graph_transforms { namespace
|
| D | inline_partitionedcall_test.cc | 27 namespace graph_transforms { namespace
|
| D | remove_attribute.cc | 26 namespace graph_transforms { namespace
|
| D | transform_utils.h | 32 namespace graph_transforms {
285 static tensorflow::graph_transforms::TransformRegistrar \
|
| D | rename_attribute.cc | 26 namespace graph_transforms { namespace
|
| D | transform_graph_test.cc | 32 namespace graph_transforms { namespace
113 graph_transforms::MapNamesToNodes(out_graph_def, &out_node_map); in TestConstantFolding()
|
| /external/tensorflow/tensorflow/python/util/ |
| D | transform_graph_wrapper.cc | 39 graph_transforms::TransformParameters params_list; in TransformGraphWithStringInputs()
40 Status parse_status = graph_transforms::ParseTransformParameters( in TransformGraphWithStringInputs()
48 Status transform_status = graph_transforms::TransformGraph( in TransformGraphWithStringInputs()
|
| /external/tensorflow/tensorflow/tools/tfg_graph_transforms/ |
| D | tfg_graph_transforms_main.cc | 139 mlir::tfg::graph_transforms::ReadModelProto<tensorflow::SavedModel>( in ImportModel()
147 mlir::tfg::graph_transforms::ReadModelProto<tensorflow::GraphDef>( in ImportModel()
162 mlir::tfg::graph_transforms::ReadModelProto<tensorflow::SavedModel>( in ExportTFGModule()
172 return mlir::tfg::graph_transforms::SerializeProto< in ExportTFGModule()
182 return mlir::tfg::graph_transforms::SerializeProto<tensorflow::GraphDef>( in ExportTFGModule()
|
| D | utils.cc | 25 namespace graph_transforms { namespace
|