- Sort Score
- Result 10 results
- Languages All
Results 71 - 80 of 88 for input_shapes_ (0.18 sec)
-
tensorflow/compiler/mlir/lite/transforms/lower_static_tensor_list.cc
size_diff, scalar_zero); // Build the argument/result types for if branch function. auto input_shape = rewriter.create<TF::ShapeOp>( loc, tensorflow::GetTypeFromTFTensorShape({-1}, shape_dtype), input_handle); Type branch_args_type[] = {input_handle.getType(), input_shape.getType(), size_diff.getType(), size.getType()}; Type branch_result_type[] = {result_type};
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jun 11 20:00:43 UTC 2024 - 70.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/tests/passes/insert_calibration_statistics_saver.mlir
func.func private @cond_false_80(%arg0: tensor<1x4xf32> {tf._user_specified_name = "x"}) -> (tensor<i1>, tensor<1x3xf32>) attributes {tf._construction_context = "kEagerRuntime", tf._input_shapes = [#tf_type.shape<1x4>], tf._original_func_name = "cond_false_8"} { %cst = "tf.Const"() <{value = dense<true> : tensor<i1>}> {device = ""} : () -> tensor<i1>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 01:09:50 UTC 2024 - 24.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/legalize_tf.cc
op->getLoc(), permuation_tensor_type, DenseElementsAttr::get(permuation_tensor_type, permute)); auto input_shape = input_type.getShape(); llvm::SmallVector<int64_t, 4> permuted_shape(input_shape.begin(), input_shape.end()); // Swaps z dimension and x dimension to get permuted shape. std::iter_swap(permuted_shape.begin() + input_rank - 1,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon May 20 20:06:54 UTC 2024 - 45.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/ir/tfl_ops.cc
Value input_permutation) { auto input_shape = input_value.getType().dyn_cast<ShapedType>().getShape(); absl::flat_hash_map<int32_t, int32_t> permutation_map; for (size_t before_dim_idx = 0, after_dim_idx = 0; before_dim_idx < input_shape.size(); ++before_dim_idx) { if (input_shape[before_dim_idx] == 1) { continue; }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 02 09:41:17 UTC 2024 - 169.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/tests/insert_custom_aggregation_ops.mlir
return %3 : tensor<1x3xf32> } func.func private @cond_false_80(%arg0: tensor<1x4xf32> {tf._user_specified_name = "x"}) -> (tensor<i1>, tensor<1x3xf32>) attributes {tf._construction_context = "kEagerRuntime", tf._input_shapes = [#tf_type.shape<1x4>], tf._original_func_name = "cond_false_8"} { %cst = "tf.Const"() <{value = dense<true> : tensor<i1>}> {device = ""} : () -> tensor<i1>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 10 04:07:09 UTC 2024 - 32.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/quantize_composite_functions.cc
llvm::ArrayRef<int64_t> input_shape = input_type.getShape(); // If weight_shape[2] != 1, it means weight shape was already restored. if (weight_shape[2] != 1) return failure(); // Weight was reshaped into [H, W, 1, InxMul]. // Since we know in_channels from input_shape, we can derive multiplier. int64_t in_channels = input_shape[3];
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 54.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/einsum.cc
int32_t num_reshape_segids, Location loc, PatternRewriter* rewriter) { // Build ShapeOp auto input_shape = rewriter->create<TF::ShapeOp>(loc, value, rewriter->getBoolAttr(true)); // Build UnsortedSegmentProdOp Type segProdresultType =
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 33.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/ir/tf_ops_a_m.cc
LogicalResult BatchToSpaceOp::verify() { BatchToSpaceOp op = *this; // Op already has a constraint that block_size >= 2. int64_t block_size = op.getBlockSize(); llvm::SmallVector<int64_t, 4> input_shape(4, ShapedType::kDynamic); auto input_type = mlir::cast<TensorType>(op.getInput().getType()); if (input_type.hasRank()) { if (input_type.getRank() != 4) return op.emitOpError()
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 146.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/tests/replace_cast_hacks_with_tf_xla_ops.mlir
// CHECK-DAG: %[[filter:.*]] = "tf.Const"() <{value = dense<2> : tensor<2x3x3x2xi8>}> {device = ""} : () -> tensor<2x3x3x2xi8> // CHECK-DAG: %[[input_shape:.*]] = "tf.Shape"({{.*}}) : (tensor<?x?x?x3xi8>) -> tensor<4xi32>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Oct 30 06:52:55 UTC 2023 - 81K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/passes.h
CreateReplicateTensorListInitOpsPass(); // Performs Shape Inference on the TensorFlow dialect using the global registry. std::unique_ptr<OperationPass<ModuleOp>> CreateTFShapeInferencePass( ArrayRef<ArrayRef<int64_t>> input_shapes = {}); // Performs TF.data optimizations. std::unique_ptr<OperationPass<func::FuncOp>> CreateTFDataOptimizationPass(); std::unique_ptr<OperationPass<func::FuncOp>> CreateMoveTransposesPass();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Jun 12 21:18:05 UTC 2024 - 31.8K bytes - Viewed (0)