- Sort Score
- Result 10 results
- Languages All
Results 1 - 6 of 6 for user_ops (0.28 sec)
-
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) -
tensorflow/compiler/mlir/tensorflow/transforms/tf_saved_model_freeze_variables.cc
Operation* user_op, int argument_index, ElementsAttr value, llvm::SmallVector<std::pair<Region*, int>, 4>* work_list, llvm::MapVector<Operation*, llvm::SmallVector<unsigned int, 4>>* arguments_to_erase) { if (auto read_variable_op = dyn_cast<TF::ReadVariableOp>(user_op)) { (*arguments_to_erase)[read_variable_op]; PropagateUsage(read_variable_op, value); } else if (auto call = dyn_cast<CallOpInterface>(user_op)) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 09:56:53 UTC 2024 - 19.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/fused_kernel_matcher.mlir
// CHECK: %[[VAL_0:.*]] = "tf._FusedConv2D"(%arg2, %arg1, %arg0) <{data_format = "NHWC", dilations = [1, 1, 1, 1], epsilon = 0.000000e+00 : f32, explicit_paddings = [], fused_ops = ["BiasAdd"], num_args = 1 : i64, operandSegmentSizes = array<i32: 1, 1, 1, 0>, padding = "SAME", strides = [1, 1, 1, 1], use_cudnn_on_gpu = true}> {TArgs = [f32]} : (tensor<8x32x32x3xf32>, tensor<1x1x3x128xf32>, tensor<128xf32>) -> tensor<*xf32>...
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Oct 30 06:52:55 UTC 2023 - 13.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/fused_kernel_matcher.cc
// together; epsilon (only with FusedBatchNorm). std::vector<NamedAttribute> attrs = contraction->getAttrs(); ArrayAttr fused_ops_attr = ArrayAttr::get(context, fused_ops); attrs.push_back( NamedAttribute(StringAttr::get(context, "fused_ops"), fused_ops_attr)); // Epsilon is used only in fusions with the FusedBatchNorm op, so we zero it // here. Attribute epsilon = rewriter.getF32FloatAttr(0);
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)