- Sort Score
- Result 10 results
- Languages All
Results 91 - 100 of 111 for _output_shapes (0.19 sec)
-
tensorflow/compiler/mlir/lite/stablehlo/transforms/legalize_hlo.cc
} auto output_shape = mlir::cast<RankedTensorType>(conv_op.getResult().getType()) .getShape(); SmallVector<int64_t, 4> transposed_output_shape = { output_shape[dnums.getOutputBatchDimension()], output_shape[dnums.getOutputSpatialDimensions().data()[0]], output_shape[dnums.getOutputSpatialDimensions().data()[1]],
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 154.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/optimize.cc
mlir::cast<ShapedType>(input.getType()).getShape(); ArrayRef<int64_t> output_shape = mlir::cast<ShapedType>(output.getType()).getShape(); int64_t agg_value = 1; for (size_t i = agg_start_idx; i < input_shape.size() - 1; ++i) { agg_value *= input_shape[i]; } return (agg_value == output_shape[agg_start_idx]); } // Returns whether the given type `a` is broadcast-compatible with `b`.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Apr 30 00:40:15 UTC 2024 - 102.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/flatbuffer_import.cc
for (auto s : new_shape) { shape.push_back( builder.getI32IntegerAttr(mlir::TFL::ConvertToTfliteSize(s))); } auto output_shape = DenseElementsAttr::get(shape_type, shape); auto shape_op = builder.create<tfl::ConstOp>(loc, output_shape); op_state.addOperands({shape_op}); } op_state.addTypes({type}); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 21 18:21:50 UTC 2024 - 66.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/side-effect-analysis-test.mlir
// expected-remark@above {{ID: 0}}
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Dec 20 04:39:18 UTC 2023 - 129.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/legalize_patterns.td
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jun 04 13:30:42 UTC 2024 - 28.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/canonicalize.mlir
%4 = "tf.Case"(%2, %arg0, %arg1) {branches = [@sub, @add], output_shapes = [#tf_type.shape<>], device = "noodle", is_stateless = false} : (tensor<i32>, tensor<f32>, tensor<f32>) -> tensor<f32> // CHECK: PartitionedCall // CHECK-SAME: f = @sub // CHECK-SAME: _cluster_launch = "not_ready" // CHECK-SAME: device = "noodle"
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 09 22:07:10 UTC 2024 - 132.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/duplicate_shape_determining_constants.cc
CompileTimeConstantOperand<TF::SegmentSumV2Op, 2>, // $num_segments CompileTimeConstantOperand<TF::SliceOp, 1, 2>, // $begin, $size CompileTimeConstantOperand<TF::SparseToDenseOp, 1>, // $output_shape CompileTimeConstantOperand<TF::SplitOp, 0>, // $split_dim // $size_splits, $split_dim CompileTimeConstantOperand<TF::SplitVOp, 1, 2>,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Mar 22 05:52:39 UTC 2024 - 17.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/tests/unfreeze_constants.mlir
%0:3 = "tf.While"(%cst_0, %cst_1, %arg0) {T = [i32, i32, f32], _lower_using_switch_merge = true, _num_original_outputs = 4 : i64, _read_only_resource_inputs = [], body = @while_body, cond = @while_cond, device = "", is_stateless = true, output_shapes = [#tf_type.shape<>, #tf_type.shape<>, #tf_type.shape<1x5x5x1024>], parallel_iterations = 10 : i64, shape_invariant} : (tensor<i32>, tensor<i32>, tensor<1x5x5x1024xf32>) -> (tensor<i32>, tensor<i32>, tensor<1x5x5x1024xf32>)
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Oct 30 06:52:55 UTC 2023 - 17.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/ir/tf_generated_ops.td
let summary = "A container for an iterator resource."; let arguments = (ins ConfinedAttr<TypeArrayAttr, [ArrayMinCount<1>]>:$output_types, ConfinedAttr<TF_ShapeAttrArray, [ArrayMinCount<1>]>:$output_shapes ); let results = (outs Res<TF_ResourceTensor, [{A handle to the iterator that can be passed to a "MakeIterator" or "IteratorGetNext" op. In contrast to Iterator, AnonymousIterator prevents
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jun 11 23:24:08 UTC 2024 - 793K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/legalize_tf.cc
Value none = rewriter.create<TFL::NoValueOp>( op->getLoc(), rewriter.getNoneType(), rewriter.getUnitAttr()); Value output_shape = CreateCastToInt32(tf_op.getInputSizes(), op->getLoc(), rewriter); rewriter.replaceOpWithNewOp<TFL::Conv3DTransposeOp>( op, tf_op.getType(), output_shape, tf_op.getFilter(), tf_op.getOutBackprop(), /*bias=*/none, dilation_depth_factor, dilation_height_factor,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon May 20 20:06:54 UTC 2024 - 45.2K bytes - Viewed (0)