- Sort Score
- Result 10 results
- Languages All
Results 1 - 4 of 4 for fused_ops (0.16 sec)
-
tensorflow/compiler/mlir/tensorflow/transforms/fused_kernel_matcher.cc
if (bias_add_op) rewriter.setInsertionPoint(bias_add_op); Value fused_op = rewriter.create<FusedOpT>(fused_loc, result_type, ValueRange(operands), attrs); auto op_to_replace = fuse_activation ? activation : bias_add; rewriter.replaceOp(op_to_replace, ValueRange({fused_op})); return success(); } }; const char kDeviceAttr[] = "device";
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 14.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/tests/legalize-tf-prefer-tf2xla.mlir
%conv2d = "tf._FusedConv2D"(%input, %filter, %bias, %act, %input_scale, %side_input_scale) { data_format = "NHWC", dilations = [1, 1, 1, 1], epsilon = 9.99999974E-5 : f32, explicit_paddings = [], filter_format = "HWIO", fused_ops = ["BiasAdd", "Relu"], leakyrelu_alpha = 2.000000e-01 : f32, num_args = 2 : i64, operandSegmentSizes = array<i32: 1, 1, 2, 2>, padding = "SAME", strides = [1, 1, 1, 1], use_cudnn_on_gpu = true
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sat Apr 06 15:32:52 UTC 2024 - 15.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/tf_tfl_translate.cc
toco_flags.set_reduce_type_precision(reduce_type_precision); // Read list of user select ops. llvm::SmallVector<llvm::StringRef, 2> user_ops; (llvm::StringRef(select_user_tf_ops)) .split(user_ops, ',', /*MaxSplit=*/-1, /*KeepEmpty=*/false); llvm::for_each(user_ops, [&toco_flags](llvm::StringRef op_name) { *(toco_flags.add_select_user_tf_ops()) = op_name.str(); }); std::string result;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 03 18:01:23 UTC 2024 - 14K bytes - Viewed (0) -
tensorflow/cc/BUILD
"manip_ops", "nn_ops", "no_op", "parsing_ops", "random_ops", "sparse_ops", "state_ops", "string_ops", "training_ops", "user_ops", ], other_hdrs = [ "ops/array_ops.h", "ops/const_op.h", "ops/math_ops.h", "ops/dataset_ops.h", "ops/experimental_dataset_ops.h",
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Mar 27 18:00:18 UTC 2024 - 23.5K bytes - Viewed (0)