Home
last modified time | relevance | path

Searched refs:GraphOptimizationPassOptions (Results 1 – 25 of 98) sorted by relevance

1234

/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/
H A Dmlir_roundtrip_pass.cc36 Import(const GraphOptimizationPassOptions& options, const Graph& graph, in Import()
55 const GraphOptimizationPassOptions& options, in Export()
61 static Status Roundtrip(const GraphOptimizationPassOptions& options, in Roundtrip()
67 Status MlirRoundtripPass::Run(const GraphOptimizationPassOptions& options) { in Run()
H A Dmlir_roundtrip_pass.h28 Status Run(const GraphOptimizationPassOptions& options) override;
/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/
H A Doptimization_registry_test.cc26 Status Run(const GraphOptimizationPassOptions& options) override { in Run()
39 GraphOptimizationPassOptions options; in TEST()
53 Status Run(const GraphOptimizationPassOptions& options) override { in Run()
71 GraphOptimizationPassOptions options; in RunPass()
H A Doptimization_registry.h38 struct GraphOptimizationPassOptions { struct
88 virtual Status Run(const GraphOptimizationPassOptions& options) = 0; argument
125 const GraphOptimizationPassOptions& options);
H A Dforward_type_inference.h45 Status Run(const GraphOptimizationPassOptions& options) override;
54 Status Run(const GraphOptimizationPassOptions& options) override;
H A Dforward_type_inference.cc129 const GraphOptimizationPassOptions& options) { in Run()
328 const GraphOptimizationPassOptions& options) { in Run()
H A Dmkl_tfconversion_pass.cc79 Status Run(const GraphOptimizationPassOptions& options);
424 Status MklToTfConversionPass::Run(const GraphOptimizationPassOptions& options) { in Run()
H A Dcontrol_flow_deps_to_chains.h71 Status Run(const GraphOptimizationPassOptions& options) override;
H A Disolate_placer_inspection_required_ops_pass.h58 Status Run(const GraphOptimizationPassOptions& options) override;
/aosp_15_r20/external/tensorflow/tensorflow/core/tpu/graph_rewrite/
H A Ddistributed_tpu_configuration_rewrite_pass.h37 Status Run(const GraphOptimizationPassOptions& options) override;
46 Status Run(const GraphOptimizationPassOptions& options) override;
H A Dencapsulate_tpu_computations_pass.h39 Status Run(const GraphOptimizationPassOptions& options) override;
63 Status Run(const GraphOptimizationPassOptions& options) override;
H A Dupdate_tpu_embedding_ops_passes.h30 Status Run(const GraphOptimizationPassOptions& options) override;
35 Status Run(const GraphOptimizationPassOptions& options) override;
H A Dupdate_tpu_embedding_ops_passes.cc49 const GraphOptimizationPassOptions& options) { in Run()
280 const GraphOptimizationPassOptions& options) { in Run()
H A Ddistributed_tpu_configuration_rewrite_pass.cc224 const GraphOptimizationPassOptions& options) { in Run()
342 const GraphOptimizationPassOptions& options) { in Run()
H A Dcombine_tpu_embedding_load_retrieve_pass.h29 Status Run(const GraphOptimizationPassOptions& options) override;
H A Dconfigure_tpu_embedding_rewrite_pass.h33 Status Run(const GraphOptimizationPassOptions& options) override;
H A Dvariable_merger_pass.h42 Status Run(const GraphOptimizationPassOptions& options) override;
/aosp_15_r20/external/tensorflow/tensorflow/compiler/jit/
H A Dmark_for_compilation_pass.h40 Status Run(const GraphOptimizationPassOptions& options) override;
43 Status RunForTest(const GraphOptimizationPassOptions& options,
H A Dtest_util.h52 GraphOptimizationPassOptions CreateGraphOptimizationPassOptions( in CreateGraphOptimizationPassOptions()
54 GraphOptimizationPassOptions options; in CreateGraphOptimizationPassOptions()
H A Dbuild_xla_ops_pass.cc297 const GraphOptimizationPassOptions& options, in ReplaceFunctionCallWithPartitionedCall()
468 const GraphOptimizationPassOptions& options, in ReplaceNodeWithXlaCompileAndXlaRun()
571 Status BuildXlaOpsPass::Run(const GraphOptimizationPassOptions& options) { in Run()
H A Dincrease_dynamism_for_auto_jit_pass.h52 Status Run(const GraphOptimizationPassOptions& options) override;
H A Dpartially_decluster_pass.h30 Status Run(const GraphOptimizationPassOptions& options) override;
H A Dcluster_scoping_pass.h33 Status Run(const GraphOptimizationPassOptions& options) override;
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/
H A Dmlir_graph_optimization_pass.h150 virtual Status Run(const GraphOptimizationPassOptions& options,
179 Status Run(const GraphOptimizationPassOptions& options) override;
/aosp_15_r20/external/tensorflow/tensorflow/core/tpu/kernels/
H A Dtpu_functional_ops.h226 const GraphOptimizationPassOptions& optimization_options,
232 const GraphOptimizationPassOptions& optimization_options, Graph* graph,

1234