- Sort Score
- Result 10 results
- Languages All
Results 51 - 60 of 71 for stateful_partitioned_call (0.33 sec)
-
tensorflow/compiler/mlir/quantization/tensorflow/quantize_preprocess.cc
pm.addNestedPass<mlir::func::FuncOp>( mlir::quant::stablehlo::CreateConvertTFQuantOpsToMHLOPass()); pm.addPass(mlir::createCanonicalizerPass()); // TF -> StableHLO legalization. // Skip StatefulPartitionedCall to preserve aliased functions. mlir::odml::AddLegalizeTFToStablehloPasses(pm, /*skip_quantization_ops=*/true, /*skip_resize=*/false,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Apr 24 12:49:45 UTC 2024 - 9.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/tests/passes/replace_stablehlo_ops_in_main_function_with_xla_call_module_ops.mlir
// CHECK: return %[[SUBGRAPH_1]] : tensor<1024x3xf32> // CHECK: } } // ----- // main function contains PartitionedCall and StatefulPartitionedCall ops which // is used to preserve aliased functions. This test make sure stablehlo ops in // each PartitionedCall functions are lifted.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 01:09:50 UTC 2024 - 39.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/tf_saved_model_optimize_global_tensors_interprocedural.mlir
func.func @f(%arg0: tensor<!tf_type.resource<tensor<f32>>> {tf_saved_model.bound_input = @v}) -> (tensor<f32> {tf_saved_model.index_path = []}) attributes {tf_saved_model.exported_names = ["f"]} { %val = "tf.StatefulPartitionedCall"(%arg0) {config = "", config_proto = "", executor_type = "", f = @f_callee} : (tensor<!tf_type.resource<tensor<f32>>>) -> (tensor<f32>) func.return %val : tensor<f32> }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Oct 31 08:49:35 UTC 2023 - 10.8K bytes - Viewed (0) -
tensorflow/c/experimental/saved_model/internal/saved_model_api_test.cc
// dtype: DT_FLOAT // tensor_shape: { // } // } // } // outputs: { // key : "output_0" // value: { // name : "StatefulPartitionedCall:0" // dtype: DT_FLOAT // tensor_shape: { // } // } // } // method_name: "tensorflow/serving/predict" // } // }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Apr 23 08:08:45 UTC 2024 - 21.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/sparsecore/embedding_sequencing.cc
} } TF::StatefulPartitionedCallOp MakeFuncCaller( mlir::OpBuilder& builder, const Location& loc, func::FuncOp func, const llvm::SetVector<Value>& operands) { // Constructs a tf.StatefulPartitionedCall to the function provided in 'func' // using the operands in 'operands'. Assumes the insertion point on builder is // already set. auto symbol =
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 39.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfrt/tests/tf_to_corert/control_flow.mlir
// CHECK: [[results_0:%.*]]:2 = tfrt.call @callee([[chain]] // CHECK-SAME: (!tfrt.chain, !tfrt_fallback.tf_tensor) -> (!tfrt.chain, !tfrt_fallback.tf_tensor) %1 = "tf.StatefulPartitionedCall"(%0) {config = "", config_proto = "", executor_type = "", f = @callee} : (tensor<i32>) -> (tensor<i32>) // CHECK-NEXT: [[results_1:%.*]]:2 = tfrt.call @callee([[chain]]
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 14 00:40:32 UTC 2024 - 17.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/tests/unfreeze_constants.mlir
func.func @serving_default(%arg0: tensor<8xf32> {tf_saved_model.index_path = ["input"]}) -> (tensor<8xf32> {tf_saved_model.index_path = ["output"]}) attributes {tf.entry_function = {control_outputs = "", inputs = "", outputs = "StatefulPartitionedCall:0"}, tf_saved_model.exported_names = ["serving_default"]} { %cst_0 = "tf.Const"() {device = "", value = dense<1.0> : tensor<8xf32>} : () -> tensor<8xf32>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Oct 30 06:52:55 UTC 2023 - 17.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/tf_to_tfl_flatbuffer.cc
return llvm::isa_and_nonnull<mlir::TF::StatefulPartitionedCallOp>(op) ? mlir::WalkResult::interrupt() : mlir::WalkResult::advance(); }); if (result.wasInterrupted()) { // StatefulPartitionedCall ops are not supported by the tflite runtime. mlir::TFL::AttachErrorCode( module.emitError( "The Graph contains unsupported `StatefulPartionedCallOp`(s), will "
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 03 18:01:23 UTC 2024 - 23.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/sparsecore/embedding_pipelining.cc
// TPUCompilationResultsOp. if (op->hasAttr(kTpuCompilationStatus)) { op->setAttr(kTpuCompilationStatus, attr); } } // A helper class to inline TF::StatefulPartitionedCall ops struct Inliner : public InlinerInterface { Inliner(OpBuilder& builder, SymbolTable& symbol_table) : InlinerInterface(builder.getContext()), builder(builder),
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 92.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/stack_ops_decomposition.cc
result.replaceAllUsesWith(if_op.getOperand(then_aliased_input + 1)); } } if_op.replaceAllUsesWith(new_if); if_op.erase(); return success(); } // Handles stack usage by a tf.StatefulPartitionedCall or a tf.PartitionedCall. // It will first check if the callee was previously handled, and try to reuse // that result if so. Otherwise, it will clone and convert the callee function,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 23.8K bytes - Viewed (0)