- Sort Score
- Result 10 results
- Languages All
Results 151 - 160 of 234 for eraseOp (0.11 sec)
-
tensorflow/compiler/mlir/lite/transforms/trim_functions_tf.cc
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Oct 07 21:08:41 UTC 2022 - 4.6K bytes - Viewed (0) -
tensorflow/compiler/jit/encapsulate_xla_computations_pass.cc
VLOG(2) << "Deleting node " << node->DebugString(); // Ensure that we do not attempt to add control edges to nodes that are // deleted. control_inputs.erase(node); control_outputs.erase(node); graph->RemoveNode(node); } TF_ASSIGN_OR_RETURN(Node * xla_launch, graph->AddNode(def)); for (int i = 0, end = data_inputs.size(); i < end; ++i) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Mar 12 06:33:33 UTC 2024 - 15.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfr/passes/decompose.cc
} // Copy all the unregisted attributes to the new op. if (failed(CopyAllowedUnregisteredAttrs(op, new_op, defined_attrs))) { return WalkResult::interrupt(); } op->erase(); changed |= true; return WalkResult::advance(); }); // If `changed` is false, it is considered as a failure, so the recursive // rewrite will stop. return success(changed); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 14.6K bytes - Viewed (0) -
platforms/core-configuration/kotlin-dsl/src/test/kotlin/org/gradle/kotlin/dsl/execution/ResidualProgramCompilerTest.kt
print("stage 2") """.replaceIndent() ) val fragment = source.fragment(0..6, 8..29) val stage1 = Program.Plugins(fragment) val stage2 = source.map { it.erase(listOf(fragment.range)) } val stagedProgram = Dynamic( Static(ApplyPluginRequestsOf(stage1), ApplyBasePlugins), stage2 ) assertStagedTopLevelProjectProgram(
Registered: Wed Jun 12 18:38:38 UTC 2024 - Last Modified: Mon Apr 01 13:09:45 UTC 2024 - 22.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/internal/passes/extract_outside_compilation.cc
// Remove unused message identification ops. if (communication_key_index == 0) for (auto op : compilation_key_ops) op->erase(); if (communication_key_index == 0 || core_to_device_ordinal.empty()) for (auto op : core_to_device_ordinal_op) op->erase(); for (mlir::tf_device::LaunchOp tmp_host_launch_op : core_to_tmp_launch) RemoveOutsideCompilation(tmp_host_launch_op);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Apr 30 21:25:12 UTC 2024 - 68.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tfg-to-tfe.cc
op->getLoc(), source_op.getToken(), /*input=*/op->getOperand(0), /*controlInputs=*/new_operands); op->replaceAllUsesWith( ValueRange({source_op.getOutput(), source_op.getControl()})); op->erase(); }); } class ConvertGraphOp : public OpConversionPattern<tfg::GraphOp> { public: using OpConversionPattern::OpConversionPattern; LogicalResult matchAndRewrite(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 17 17:58:54 UTC 2024 - 21.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tpu_parallel_execute_sink_resource_write.cc
for (auto result : llvm::zip(results_to_remap, new_parallel_execute.getResults())) std::get<0>(result).replaceAllUsesWith(std::get<1>(result)); parallel_execute.erase(); } void TPUParallelExecuteSinkResourceWrite::runOnOperation() { llvm::SmallVector<tf_device::ParallelExecuteOp, 4> parallel_executes; getOperation().walk([&](tf_device::ParallelExecuteOp parallel_execute) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Dec 06 04:46:18 UTC 2022 - 6.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/initialize_variables_in_session_init.cc
builder.getContext(), kSessionInitFuncName))); if (session_init_op) { session_init_op->replaceAllUsesWith(new_session_init_op); session_init_op->erase(); } return func; } func::FuncOp GetOrCreateSessionInitFunc(ModuleOp module) { SessionInitializerOp session_init_op = GetSessionInitializerOp(module);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 23 09:05:47 UTC 2024 - 7K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/quantization/stablehlo/quantization.cc
absl::flat_hash_map<std::string, SignatureDef> signature_def_map( signatures.begin(), signatures.end()); // Init op is only used during initialization and it's not a target for // quantization. signature_def_map.erase(kSavedModelInitOpSignatureKey); return signature_def_map; } // Retrieves the function name -> function alias mapping from the // `SavedModelBundle`. // TODO: b/314124142 - Remove the need for this function.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sun Apr 14 10:49:12 UTC 2024 - 7.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/legalize_stablehlo_composite_to_tfl_custom.cc
// Build TFL custom op, replace composite with custom op. TFL::CustomOp tfl_custom_op = BuildCustomOp(composite, custom_option_buffer); composite->replaceAllUsesWith(tfl_custom_op); composite->erase(); }); } }; static PassRegistration<LegalizeCompositeToCustomOpPass> pass; } // namespace odml
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 6.8K bytes - Viewed (0)