- Sort Score
- Result 10 results
- Languages All
Results 31 - 40 of 52 for _output_shapes (0.18 sec)
-
tensorflow/compiler/mlir/tensorflow/tests/graphdef2mlir/tf-data-pipeline.pbtxt
op: "TensorSliceDataset" input: "tensors/normalize_tensors/component_0" attr { key: "Toutput_types" value { list { type: DT_INT32 } } } attr { key: "output_shapes" value { list { shape { } } } } } node { name: "MapDataset" op: "MapDataset" input: "TensorSliceDataset" attr { key: "Targuments"
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jul 29 04:41:05 UTC 2021 - 4K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/composite_utils.cc
std::array<int64_t, 4> output_shape; // NHWC <- NCHW output_shape[0] = composite_result_shape[0]; output_shape[1] = composite_result_shape[2]; output_shape[2] = composite_result_shape[3]; output_shape[3] = composite_result_shape[1]; auto input_type = mlir::cast<ShapedType>(old_op->getOperand(0).getType()); return RankedTensorType::get(output_shape, input_type.getElementType()); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 29 18:33:05 UTC 2024 - 3.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/graphdef2mlir/shape-attrs.pbtxt
} } attr { key: "index" value { i: 4 } } } node { name: "IteratorGetNext" op: "IteratorGetNext" input: "args_0" attr { key: "output_shapes" value { list { shape { dim { size: 1 } dim { size: 8 } } shape { dim {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Dec 04 18:02:53 UTC 2020 - 5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/mlir2graphdef/case.mlir
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Mar 28 12:06:33 UTC 2022 - 2.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/quantize_patterns.td
// Transpose conv supports hybrid computation with quantized weights. def FoldQuantWeightsIntoTposeConv : Pat< (TFL_TransposeConvOp $output_shape, (TFL_DequantizeOp $quant_weights), $quant_input, $bias, $padding, $stride_h, $stride_w, $faf), (TFL_TransposeConvOp $output_shape, $quant_weights, $quant_input, $bias, $padding, $stride_h, $stride_w, $faf),
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 28 23:10:13 UTC 2024 - 2.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/mlir2graphdef/functional-if-ops.mlir
// cases, the additional attribute should be dropped. // CHECK: name: "StatefulIf" // CHECK-NOT: name: // CHECK: op: "If" // CHECK-NOT: is_stateless // CHECK: attr { // CHECK: key: "output_shapes" // CHECK: value { // CHECK: list { // CHECK: shape { // CHECK: dim { // CHECK: size: 4 // CHECK: } // CHECK: } // CHECK: } // CHECK: }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Mar 25 12:28:56 UTC 2022 - 2.5K 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/mlir/tensorflow/transforms/tf_data_optimization.td
def FuseMapAndBatch : Pat< (TF_BatchDatasetV2Op (TF_MapDatasetOp $input_dataset, $other_arguments, $f, $output_types, $output_shapes, $use_inter_op_parallelism, $preserve_cardinality, $force_synchronous, $map_dataset_metadata), $batch_size, $drop_remainder, $parallel_copy, $batch_output_types,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jun 11 23:24:08 UTC 2024 - 1.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/graphdef2mlir/graph-func-attr.pbtxt
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Nov 11 19:14:04 UTC 2020 - 2.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/tf_device_index_selector.mlir
%1 = "tf.DeviceIndex"() {device = "", device_names = ["CPU", "GPU"]} : () -> tensor<i32> %4 = "tf.Case"(%1, %arg0, %arg1) {branches = [@sub, @add], output_shapes = [#tf_type.shape<>], is_stateless = false} : (tensor<i32>, tensor<f32>, tensor<f32>) -> tensor<f32> func.return %0, %4 : tensor<i32>, tensor<f32> }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Mar 24 05:47:26 UTC 2022 - 1.1K bytes - Viewed (0)