- Sort Score
- Result 10 results
- Languages All
Results 11 - 20 of 157 for op_names (1.24 sec)
-
tensorflow/compiler/mlir/tensorflow/tests/lower_tf.mlir
// CHECK: %[[RESULT1:.*]] = "tf.Abs"(%[[RESULT0]]) // CHECK: %[[RESULT2:.*]] = "tf.Log"(%[[RESULT1]]) // CHECK: return %[[RESULT2]] %0 = "tf._UnaryOpsComposition"(%arg0) {op_names = ["Asin", "Abs", "Log"]} : (tensor<4xf32>) -> tensor<4xf32> func.return %0 : tensor<4xf32> } // CHECK-LABEL: @round_int func.func @round_int(%arg0: tensor<2xi32>) -> tensor<2xi32> {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Jan 05 18:35:42 UTC 2024 - 92K bytes - Viewed (0) -
tensorflow/c/eager/c_api_unified_experimental_graph.cc
private: friend class GraphContext; // For access to op_. TF_Graph* g_; std::unique_ptr<TF_OperationDescription> op_; // Hold `op_type` and `op_name` till both are available since we need both // to build a graph operation. string op_type_; const char* op_name_ = nullptr; // TODO(srbs): Use this. string device_name_; }; // GraphContext wraps a TF_Graph modeling a single function and manages the
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Mar 12 20:00:09 UTC 2024 - 15.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/tfl_stablehlo_pass.cc
namespace odml { static bool isDenseI64Array(llvm::StringRef op_name, llvm::StringRef field_name) { if (op_name == "stablehlo.broadcast" && field_name == "broadcast_sizes") return true; if (op_name == "stablehlo.broadcast_in_dim" && field_name == "broadcast_dimensions") return true; if ((op_name == "stablehlo.convolution" || op_name == "stablehlo.dynamic_conv") &&
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Jan 24 06:08:43 UTC 2024 - 10.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/quantization/tools/tflite_op_coverage_spec_getters_gen.cc
// Conditions when using TOCO. if (absl::StrContains(op_name, "LSTMOp")) continue; } else { // Conditions when using MLIR. if (!(absl::StrContains(op_name, "LSTMOp") || absl::StrContains(op_name, "SoftmaxOp") || absl::StrContains(op_name, "LogisticOp") || absl::StrContains(op_name, "L2NormalizationOp") ||
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Sep 06 19:12:29 UTC 2023 - 12.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/c/c_api_unified_experimental_mlir.cc
attrs_[attr_name] = TypeAttr::get(mlir_type); return absl::OkStatus(); } Status MlirAbstractOp::SetOpName(const char* const op_name) { // TODO(aminim): should we use a location? if (op_name_) { return FailedPrecondition("SetOpName called on already built op."); } op_name_ = op_name; return absl::OkStatus(); } Status MlirAbstractOp::AddRef(Type type, Type* output_type) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 28.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/metrics/error_collector_inst.cc
// Find the op names with tf or tfl dialect prefix, Ex: "tf.Abs" or "tfl.Abs". auto collectOps = [this](Operation *op) { const auto &op_name = op->getName().getStringRef().str(); if (absl::StartsWith(op_name, "tf.") || absl::StartsWith(op_name, "tfl.")) { loc_to_name_.emplace(op->getLoc(), op_name); } }; for (auto ®ion : module->getRegions()) { region.walk(collectOps); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Mar 25 01:48:36 UTC 2024 - 5.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/converter_gen.cc
static inline std::string GetOperatorBuilderName(StringRef op_name) { assert(op_name.starts_with("TFL_") && "unexpected op prefix"); assert(op_name.ends_with("Op") && "unexpected op suffix"); // E.g., AddOp -> CreateAddOperator std::ostringstream oss; oss << "Create" << op_name.drop_front(4).str() << "erator"; return oss.str(); } static inline bool IsLstmOp(const StringRef op_name) { return op_name.take_back(6) == "LSTMOp"; }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Dec 19 15:05:28 UTC 2023 - 23.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfr/python/op_reg_gen.py
all_dec_args.update(kw_dec_args) op_name = all_dec_args['op_name'] op_def = op_def_registry.get(op_name) if op_def: if len(all_dec_args) > 1: # Op has been registered, so it is a user error to specify op def. raise ValueError('Op has been registered: ' + op_name) else: # Op has been registered, then we don't need to generate register code.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Sep 28 21:37:05 UTC 2021 - 5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/utils/export_utils.cc
if (!op_name_or_status.ok()) return false; auto op_name = std::move(op_name_or_status).value(); if (op_name == "NextIteration") return mlir::isa<mlir::TF::TensorFlowRefType>( mlir::getElementTypeOrSelf(op->getOperand(0).getType())); if (op_name == "Enter" || op_name == "Exit" || op_name == "Switch" || op_name == "Merge") { return mlir::isa<mlir::TF::TensorFlowRefType>(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 17 17:58:54 UTC 2024 - 19.7K bytes - Viewed (0) -
tensorflow/c/kernels_test.cc
const char* node_name = "SomeNodeName"; const char* op_name = "FooOp"; const char* device_name = "FakeDeviceName1"; REGISTER_OP(op_name) .Input("input1: double") .Input("input2: uint8") .Output("output1: uint8") .Attr("SomeDataTypeAttr: type"); TF_KernelBuilder* builder = TF_NewKernelBuilder( op_name, device_name, &MyCreateFunc, &MyComputeFunc, &MyDeleteFunc); {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Sep 06 19:12:29 UTC 2023 - 50.4K bytes - Viewed (0)