- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 11 for ExpandDimsOp (0.31 sec)
-
tensorflow/compiler/mlir/lite/transforms/dilated_conv.h
Operation* consumer_op = maybeConsumer.second; TF::ExpandDimsOp expand_op; TF::SqueezeOp squeeze_op; int64_t expand_axis = -1; // Expand + Squeeze op. if (llvm::isa<TF::ExpandDimsOp>(producer_op)) { if (!llvm::isa<TF::SqueezeOp>(consumer_op)) { // Expand/Squeeze op must come in pair. return rewriter.notifyMatchFailure( op, "ExpandDimsOp and SqueezeOp should come in pair"); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 20K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/ops/tf_op_quant_spec.cc
// has one result operand. return isa<TF::IdentityOp, TF::CastOp, TF::ReshapeOp, TF::XlaShardingOp, TF::GatherOp, TF::GatherV2Op, TF::XlaGatherOp, TF::ExpandDimsOp, TF::SqueezeOp, TF::TransposeOp>(op); } bool IsOpWithQuantizableTrait(Operation* op) { // Supported quantizable ops. return isa<TF::XlaConvV2Op, TF::XlaDotV2Op, TF::MatMulOp, TF::Conv2DOp,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 6.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/lower_static_tensor_list.cc
Value scalar_zero = CreateI32SplatConst(loc, rewriter, {}, 0); RankedTensorType vector_type = tensorflow::GetTypeFromTFTensorShape({1}, shape_dtype); auto expanded_start_index = rewriter->create<TF::ExpandDimsOp>( loc, vector_type, start_index, scalar_zero); auto start_position = rewriter->create<TF::ConcatOp>( loc, position_type, scalar_zero, ArrayRef<Value>({expanded_start_index, partial_start_position}));
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/tfrt/analysis/cost_analysis.cc
// // TODO(chky): Find a more scalable way to figure out cheap ops. if (llvm::isa<mlir::TF::ShapeOp, mlir::TF::StridedSliceOp, mlir::TF::ReshapeOp, mlir::TF::ExpandDimsOp>(op)) { cost_map_[op] = kDefaultCheapCost; return; } // For other ops, use the sum of input sizes as its cost. int64_t cost = kDefaultCheapCost; for (auto operand : op->getOperands()) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 7.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/transforms/legalization_op_config.cc
TypeID::get<TF::DepthwiseConv2dNativeOp>(), TypeID::get<TF::DivOp>(), TypeID::get<TF::DynamicStitchOp>(), TypeID::get<TF::_EagerConstOp>(), TypeID::get<TF::EmptyOp>(), TypeID::get<TF::ExpandDimsOp>(), TypeID::get<TF::FakeQuantWithMinMaxVarsOp>(), TypeID::get<TF::FillOp>(), TypeID::get<TF::FusedBatchNormOp>(), TypeID::get<TF::FusedBatchNormGradOp>(), TypeID::get<TF::FusedBatchNormGradV2Op>(),
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Apr 24 04:08:35 UTC 2024 - 21.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/prepare_tf.cc
r++; } } auto expaned_input_type = tensorflow::GetTypeFromTFTensorShape( expanded_input_shape, input_type.getElementType()); TF::ExpandDimsOp expanded_input = rewriter.create<TF::ExpandDimsOp>( rfft_op.getLoc(), expaned_input_type, input, minus_two->getResult()); // Expanded fft_len. auto one_attr = mlir::DenseIntElementsAttr::get(one_ele_type, {1});
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 28 21:49:50 UTC 2024 - 64.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/lower_tf.cc
public: explicit LowerPackOp(MLIRContext *context) : RewritePattern( PackOp::getOperationName(), 1, context, {ConstOp::getOperationName(), ConcatV2Op::getOperationName(), ExpandDimsOp::getOperationName()}) {} LogicalResult matchAndRewrite(Operation *src_op, PatternRewriter &rewriter) const override { auto op = cast<PackOp>(src_op);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 74.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/ir/tf_ops_a_m.cc
MLIRContext* context) { results.add(flipComatibleShapeError<NotEqualOp>); } //===----------------------------------------------------------------------===// // ExpandDimsOp //===----------------------------------------------------------------------===// Type InferExpandDimsOpType(Value input, Value dim) { Type element_ty = mlir::cast<TensorType>(input.getType()).getElementType();
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/tf2xla/transforms/legalize_tf.cc
return success(); } }; class ConvertDynamicExpandDimsOp : public OpRewritePattern<TF::ExpandDimsOp> { public: using OpRewritePattern::OpRewritePattern; LogicalResult matchAndRewrite(TF::ExpandDimsOp op, PatternRewriter &rewriter) const override { auto input = op.getInput();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jun 11 20:00:43 UTC 2024 - 291.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/uniform_quantized_stablehlo_to_tfl_pass.cc
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Apr 22 09:00:19 UTC 2024 - 99.8K bytes - Viewed (0)