- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 162 for output_types (0.23 sec)
-
tensorflow/compiler/mlir/lite/utils/lstm_utils_test.cc
auto output_types = fused_lstm_func_.getFunctionType().getResults(); SmallVector<int64_t, 2> output_shape{1, mlir::ShapedType::kDynamic}; EXPECT_EQ(mlir::cast<RankedTensorType>(output_types[0]).getShape().size(), output_shape.size()); for (int i = 0; i < output_shape.size(); i++) { EXPECT_EQ(mlir::cast<RankedTensorType>(output_types[0]).getDimSize(i), output_shape[i]);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 10K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/passes/quantization_patterns.h
} // Collect all the quantized outputs and replace them by the results of // the new quantized op. llvm::SmallDenseMap<Value, int> outputs_replaced; SmallVector<Type, 4> output_types; output_types.reserve(candidate_op->getNumResults()); for (const auto& enumerated_result : llvm::enumerate(candidate_op->getResults())) { Value result = enumerated_result.value();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 10.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/tf_data_fuse_map_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<!tf_type.variant> %5 = "tf.BatchDatasetV2"(%4, %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: Mon Oct 30 06:52:55 UTC 2023 - 1.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/host_runtime/tpu_merge_variables_with_execute.cc
// (inclusive) to `end` index region (exclusive) to `output_types` and returns // the number of types added. int AppendTypes(llvm::SmallVectorImpl<Type>* output_types, tf_device::ParallelExecuteOp parallel_execute, int start, int end) { const int size_before = output_types->size(); for (int index = start; index < end; ++index) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Feb 29 17:52:11 UTC 2024 - 27K 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/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)