- Sort Score
- Result 10 results
- Languages All
Results 11 - 14 of 14 for TF_Op (0.04 sec)
-
tensorflow/compiler/mlir/tfr/integration/tfr_decompose_ctx.cc
op_state.addOperands(func.getArguments()); op_state.addTypes(output_tys); op_state.addAttributes(attrs); mlir::Operation* tf_op = op_builder.create(op_state); op_builder.create<mlir::func::ReturnOp>(loc, tf_op->getResults()); // Run the decompose passes on the module TF_RETURN_IF_ERROR(DecomposeGraph(module)); // Export the result as a FunctionDef.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Apr 29 02:34:43 UTC 2024 - 9.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/prepare_tf.cc
// * The 'dilations' attribute is optional, but it must be of the form // [1, X, Y, 1] if exists. TFConvOpType tf_op = cast<TFConvOpType>(op); if (!TFTypeIsFloat32Tensor(tf_op.getInput()) && !(allow_bf16_and_f16_type_legalization_ && TFTypeIsBFloat16OrHalfTensor(tf_op.getInput()))) return failure(); if (!TFDataFormatIsNHWC(op)) return failure(); IntegerAttr height, width;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 28 21:49:50 UTC 2024 - 64.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/ir/tf_op_base.td
//===----------------------------------------------------------------------===// // TensorFlow op definitions //===----------------------------------------------------------------------===// class TF_Op<string mnemonic, list<Trait> traits = []> : Op<TensorFlowDialect, mnemonic, traits>; //===----------------------------------------------------------------------===// // TensorFlow attribute definitions
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Oct 30 06:52:55 UTC 2023 - 30.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/BUILD
), ], tblgen = "@llvm-project//mlir:mlir-tblgen", td_file = "ir/tf_ops.td", deps = [ ":tensorflow_ops_td_files", ], ) # We only shard tf_op on name for build performance reasons. tf_ops_category_list = [ { "name": "ops_a_m", "include": "tf.[A-M].*$", }, { "name": "ops_n_z", "include": "tf.[N-Z].*$",
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 22 21:28:13 UTC 2024 - 47.7K bytes - Viewed (0)