- Sort Score
- Result 10 results
- Languages All
Results 41 - 50 of 111 for _output_shapes (0.16 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/c/c_api_experimental_test.cc
CHECK_EQ(TF_OK, TF_GetCode(status_)) << TF_Message(status_); CHECK_EQ(output_shapes->num_items, 1); int num_dims = output_shapes->items[0].num_dims; int64_t* dims = output_shapes->items[0].dims; if (!expected_shape.has_value()) { EXPECT_EQ(num_dims, -1); EXPECT_EQ(dims, nullptr); return; } EXPECT_EQ(num_dims, expected_shape->size());
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jan 17 22:27:52 UTC 2023 - 13.1K 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/utils/convert_tensor_test.cc
PartialTensorShape output_shape = ConvertTypeToTensorShape(mlir::UnrankedTensorType::get(b.getF32Type())); EXPECT_TRUE(output_shape.IsIdenticalTo(PartialTensorShape())); } TEST(ConvertTypeToTensorTypeTest, NonFullyDefinedRankedTensorType) { mlir::MLIRContext context; RegisterDialects(context); mlir::Builder b(&context); PartialTensorShape output_shape = ConvertTypeToTensorShape(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 10.4K 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/lite/utils/lstm_utils.cc
return failure(); // Build the lstm op. SmallVector<int64_t, 3> output_shape; if (time_majored) { output_shape = {time, batch, n_output}; } else { output_shape = {batch, time, n_output}; } auto result_type = mlir::RankedTensorType::get( output_shape, mlir::cast<RankedTensorType>(final_inputs.getType()).getElementType());
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 36.2K bytes - Viewed (0)