- Sort Score
- Result 10 results
- Languages All
Results 91 - 100 of 1,054 for ShapeN (0.09 sec)
-
tensorflow/compiler/mlir/tensorflow/translate/mlir_roundtrip_flags.cc
TF_RETURN_IF_ERROR(ParseNodeShapes(shapes, node_shapes)); return ParseInputArrayInfo(node_names, node_dtypes, node_shapes, inputs); } static absl::StatusOr<std::vector<int>> ParseShapeStr( absl::string_view node_shapes_str) { std::vector<int> dims; for (absl::string_view dim_str : absl::StrSplit(node_shapes_str, ',')) { // Treats empty input shape as scalar if (dim_str.empty()) continue;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 01 11:17:36 UTC 2024 - 10.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/graphdef2mlir/arg-as-fetch.pbtxt
# RUN: tf-mlir-translate -graphdef-to-mlir -tf-enable-shape-inference-on-import=false %s --mlir-print-debuginfo --print-after-all -tf-input-arrays=arg -tf-input-data-types=DT_INT32 -tf-input-shapes=8 -tf-output-arrays=arg -o - | FileCheck %s node { name: "arg" op: "_Arg" attr { key: "T" value { type: DT_INT32 } } attr { key: "index" value { i: 0 } } } versions { producer: 27
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 25 01:00:57 UTC 2021 - 775 bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/tests/end2end/quant_stats.pbtxt
# CHECK-NEXT: shape: [ 4 ], # CHECK-NEXT: type: UINT8, # CHECK-NEXT: buffer: 1, # CHECK-NEXT: name: "input0", # CHECK-NEXT: quantization: { # CHECK-NEXT: scale: [ 0.015686 ], # CHECK-NEXT: zero_point: [ 128 ] # CHECK-NEXT: }, # CHECK-NEXT: has_rank: true # CHECK-NEXT: }, { # CHECK-NEXT: shape: [ 4 ], # CHECK-NEXT: type: UINT8,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 02 09:41:17 UTC 2024 - 4K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/tests/end2end/control_flow_v1.pbtxt
} } attr { key: "shape" value { shape { dim { size: -1 } dim { size: 2 } } } } } node { name: "Placeholder_1" op: "Placeholder" attr { key: "dtype" value { type: DT_BOOL } } attr { key: "shape" value { shape { } }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jun 23 21:23:31 UTC 2020 - 3.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/api/v2/legalize_tf.h
// tpu::FunctionToHloArgs but this is deprecated. arg_shapes - The shapes of // the arguments in module_op. device_type - The device type to compile for. // use_tuple_args - Pack the incoming arg shapes into a single tuple. // custom_legalization_passes - Extra passes to lower from TF -> MHLO. // arg_shapes - The shapes of the args. // arg_core_mapping - Which args go on which cores.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sat Apr 13 07:32:57 UTC 2024 - 2.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/translate/tf_mlir_translate_cl.cc
"the import graph is used if it is skipped."), llvm::cl::init("")); // NOLINTNEXTLINE opt<std::string> input_shapes( "tf-input-shapes", llvm::cl::desc( "Input tensor shapes. Shapes for different tensors are separated by " "':', and dimension sizes for the same tensor are separated by ','"), llvm::cl::init("")); // NOLINTNEXTLINE
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Aug 10 20:59:50 UTC 2023 - 5.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow_to_stablehlo/README.md
* `--tags`: Comma-separated list of tags for loading SavedModel. Ignored for MLIR input. * `--input-arg-shapes`: A string representation of input argument shapes for 'main' entry-point, separating tensors with ':', dimension with ',', and using '?' for unknown sizes. For example, `input-arg-shapes=1,2::1,?` expresses argument shapes `[1,2]`, `[]` and `[1,?]`. * `--e`: Elide large elements attrs while dumping the output StableHLO.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 21 22:58:42 UTC 2024 - 4.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/graphdef2mlir/parse_example_v2.pbtxt
# RUN: tf-mlir-translate -graphdef-to-mlir -tf-enable-shape-inference-on-import=false %s -tf-input-arrays=input0 -tf-input-data-types=DT_STRING -tf-input-shapes=32 -tf-output-arrays=ParseExample/ParseExampleV2:0,ParseExample/ParseExampleV2:7 -o - | FileCheck %s # CHECK: %[[parse_example:.*]]:8, %[[parse_example_control:.*]] = tf_executor.island wraps "tf.ParseExampleV2"(%arg0, # CHECK: resultSegmentSizes = array<i32: 2, 2, 2, 2, 0, 0>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Aug 14 15:35:49 UTC 2023 - 3.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/utils/utils.td
def OpHasSameStaticShapes : Constraint<OpHasSameStaticShapesPred, "op must have static same input shapes">; def OpHasNotSameStaticShapes : Constraint<Neg<OpHasSameStaticShapesPred>, "op must have not static same input shapes">; def TransposeFCLastTwoDims: NativeCodeCall<"TransposeLastTwoDims($0[0].getType())">; def AreLastTwoDimsTransposed : Constraint<CPred< "TFL::AreLastTwoDimsTransposed($0)">>;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Apr 30 00:40:15 UTC 2024 - 4.8K bytes - Viewed (0) -
tensorflow/cc/gradients/data_flow_grad.cc
int32_t num_partitions; TF_RETURN_IF_ERROR( GetNodeAttr(op.node()->attrs(), "num_partitions", &num_partitions)); // Note: the shape of the partitions is a prefix of the data shape. // shape(partitions) = [5] auto partitions_shape = Shape(scope, partitions); // We now create a partitions-shaped tensor with integers from // [0..size(partitions)) This will be dynamic_partitioned with the
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sat Jul 24 13:40:35 UTC 2021 - 5.8K bytes - Viewed (0)