- Sort Score
- Result 10 results
- Languages All
Results 11 - 20 of 184 for output_types (0.21 sec)
-
tensorflow/compiler/mlir/tensorflow/transforms/group_by_dialect.cc
computeInputsOutputs(ops, &inputs, &outputs); std::vector<Type> input_types; std::vector<Type> output_types; input_types.reserve(inputs.size()); for (Value v : inputs) { input_types.push_back(v.getType()); } output_types.reserve(outputs.size()); for (Value v : outputs) { output_types.push_back(v.getType()); } // Create the function. MLIRContext* context = ops[0]->getContext();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Mar 17 07:31:01 UTC 2023 - 8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfr/passes/raise_to_tf.cc
SmallVectorImpl<Type>* output_types) const; // Creates the TF op and also the necessary tfr.cast ops to replace the // original TFR call op. LogicalResult CreateAndReplaceOp( PatternRewriter& rewriter, CallOp call_op, const SmallVectorImpl<Type>& output_types, const SmallVectorImpl<Value>& inputs, const NamedAttrList& attr_list,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 21.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/tf_data_fuse_pmap_and_batch.mlir
output_shapes = [#tf_type.shape<>], output_types = [i32], preserve_cardinality = false, sloppy = false, use_inter_op_parallelism = true, metadata = ""} : (tensor<*x!tf_type.variant>, tensor<i32>) -> tensor<!tf_type.variant> %6 = "tf.BatchDatasetV2"(%5, %0, %1) {device = "", output_shapes = [#tf_type.shape<>], output_types = [i32], parallel_copy = false,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Mar 24 05:47:26 UTC 2022 - 1.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/utils/parallel_execute_util.cc
tf_device::ParallelExecuteOp BuildParallelExecuteOp( tf_device::ClusterFuncOp cluster_func, OpBuilder* builder) { const auto output_types = cluster_func.getResultTypes(); builder->setInsertionPoint(cluster_func); auto parallel_execute = builder->create<tf_device::ParallelExecuteOp>( cluster_func.getLoc(), 1, output_types); cluster_func->remove(); auto& block = parallel_execute.GetRegionBlockWithIndex(0);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jun 13 03:57:18 UTC 2023 - 2.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfr/tests/end2end.mlir
%batch_size: i64 {tfr.name="batch_size"}, %num_parallel_calls: i64 {tfr.name="num_parallel_calls"}, %drop_remainder: i1 {tfr.name="drop_remainder"}, %f: !tfr.attr {tfr.name="func"}, %output_types: !tfr.attr {tfr.name="output_types"}, %output_shapes: !tfr.attr {tfr.name="output_shapes"}, %preserve_cardinality: i1 {tfr.name="preserve_cardinality", tfr.default=false}) -> !tfr.tensor {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Oct 30 06:52:55 UTC 2023 - 13.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/common/lift_as_function_call.cc
const StringRef func_name, const TypeRange output_types, const ValueRange args) { TF::PartitionedCallOp call_op = builder.create<TF::PartitionedCallOp>( location, output_types, args, FlatSymbolRefAttr::get(builder.getStringAttr(func_name)), /*config=*/"", /*config_proto=*/"", /*executor_type=*/"");
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 17 17:58:54 UTC 2024 - 21.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/post_quantize.cc
Value dequantized_result = dequantize_op.getInput(); output_types.push_back(dequantized_result.getType()); terminator->setOperand(i, dequantized_result); returned_op->erase(); } else { output_types.push_back(returned_value.getType()); } } auto new_func_type = builder.getFunctionType(input_types, output_types); func.setType(new_func_type); } enum RemoveVolatileOpsType {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 17.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/tests/lift_tflite_flex_ops.mlir
} : (tensor<!tf_type.variant>) -> tensor<!tf_type.variant> func.return %0 : tensor<!tf_type.variant> // CHECK: "tf.MapDataset"( // CHECK-SAME: <{f = @{{.*}}, metadata = "", output_shapes = [#tf_type.shape<>], output_types = [!tf_type.string], preserve_cardinality = true, use_inter_op_parallelism = true}> {Targuments = []} } // CHECK-LABEL: TfTakeWhileDataset
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Oct 30 06:52:55 UTC 2023 - 6.1K bytes - Viewed (0) -
tensorflow/compiler/jit/encapsulate_xla_computations_pass.cc
} // Outputs. const int num_outputs = launch->output_types().size(); absl::flat_hash_set<Node*> control_outputs; std::vector<std::vector<std::pair<Node*, int>>> data_outputs(num_outputs); const DataTypeVector& output_types(launch->output_types()); for (const Edge* le : launch->out_edges()) { if (le->IsControlEdge()) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Mar 12 06:33:33 UTC 2024 - 15.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/utils/lstm_utils.cc
size_of_splits = dim / splits; output_shape.push_back(size_of_splits); } else { output_shape.push_back(dim); } } SmallVector<mlir::Type, 4> output_types; for (int i = 0; i < splits; ++i) { output_types.push_back( mlir::RankedTensorType::get(output_shape, input_type.getElementType())); } auto size_of_splits_op = Create1DConstantOp(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 36.2K bytes - Viewed (0)