Search Options

Results per page
Sort
Preferred Languages
Advance

Results 31 - 40 of 1,217 for OP (0.03 sec)

  1. tensorflow/compiler/mlir/tf2xla/internal/passes/mark_ops_for_outside_compilation.cc

      if (op.getDialect() != tf_dialect) return true;
      // Assert has a legalization that later removes it so we don't want to outside
      // compile it ever for performance reasons.
      if (llvm::isa<mlir::TF::AssertOp>(op)) return true;
    
      if (HasStringOperand(op)) return false;
      if (HasStringResult(op)) return false;
      if (MatchesPattern(op, supported_ops)) return true;
    
      auto abstractOp = op.getRegisteredInfo();
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 21.4K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/lite/ir/tfl_ops.cc

    mlir::LogicalResult ReshapeOp::verify() {
      ReshapeOp op = *this;
      auto error_handler = [&op](const llvm::Twine& message) -> LogicalResult {
        return op.emitOpError() << message;
      };
      TensorType expected_ty;
      if (failed(GetReshapeOutputType(op.getInput(), op.getShape(), error_handler,
                                      expected_ty)))
        return failure();
    
      auto output_ty = op.getType().dyn_cast<RankedTensorType>();
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu May 02 09:41:17 UTC 2024
    - 169.2K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/quantization/stablehlo/passes/nchw_convolution_to_nhwc.cc

            op.getLoc(), /*resultType0=*/new_conv_output_tensor_type,
            /*lhs=*/input_transpose_op,
            /*rhs=*/filter_transpose_op,
            /*window_strides=*/op.getWindowStridesAttr(),
            /*padding=*/op.getPaddingAttr(),
            /*lhs_dilation=*/op.getLhsDilationAttr(),
            /*rhs_dilation=*/op.getRhsDilationAttr(),
            /*window_reversal=*/op.getWindowReversalAttr(),
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 8.2K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/tfr/passes/decompose.cc

                  builder.create<ConstOp>(op->getLoc(), output_type, attribute);
            } else {
              attr_cst = builder.create<mlir::arith::ConstantOp>(
                  op->getLoc(), cast<TypedAttr>(attribute));
            }
            new_operands.push_back(attr_cst);
          }
        }
    
        // Create the TFR call op
        auto new_op = builder.create<CallOp>(
            op->getLoc(), compose_func_type.getResults(),
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 14.6K bytes
    - Viewed (0)
  5. tensorflow/compiler/mlir/quantization/stablehlo/passes/quantization_patterns.cc

      // op involves a region, the op is only quantizable when the region only
      // consists of ops whose quantization parameters can be propagated from
      // outside.
      bool IsNestedRegionQuantizable(Operation* op) const {
        for (Region& region : op->getRegions()) {
          for (Operation& op : region.getOps()) {
            if (!IsOpQuantizableInNestedRegion(op)) {
              return false;
            }
          }
        }
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri May 03 06:04:36 UTC 2024
    - 41.7K bytes
    - Viewed (0)
  6. tensorflow/compiler/mlir/tf2xla/transforms/legalize_tf_patterns.td

    // conversion, use the first input to get inputs rank. Other inputs need not be
    // ranked.
    // Defining op for `axis` is TensorFlow constant op in the pattern as during
    // the conversion, original Concat op operands still refers to the old ops even
    // if HLO constant op is introduced as an replacement for the TensorFlow
    // Constant op.
    def : Pat<(TF_ConcatV2Op $inputs, (ConstantLikeMatcher OneElementAttr:$axis)),
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon May 06 18:46:23 UTC 2024
    - 34.8K bytes
    - Viewed (0)
  7. tensorflow/compiler/mlir/tensorflow/transforms/tfg-to-tfe.cc

      block.walk([&](Operation *op) {
        if (op->getName().getStringRef() != "tfg.NextIteration") return;
        mlir::OpBuilder builder(op);
    
        llvm::SmallVector<Value, 2> new_operands;
        FilterOutBlockArgControlDep(op->getOperands().drop_front(), new_operands);
    
        auto source_op = builder.create<tf_executor::NextIterationSourceOp>(
            op->getLoc(), op->getOperand(0).getType());
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri May 17 17:58:54 UTC 2024
    - 21.2K bytes
    - Viewed (0)
  8. tensorflow/compiler/mlir/tensorflow/transforms/sparsecore/embedding_sequencing.cc

    }
    
    bool IsTPUOp(mlir::Operation* op) {
      return op->hasAttr(TF::kReplicationInfoAttr);
    }
    
    StringAttr GetReplicationAttr(mlir::Operation* op) {
      return op->getAttrOfType<StringAttr>(TF::kReplicationInfoAttr);
    }
    
    StringAttr GetReplicationAttr(TF::TPUCompilationResultOp op) {
      // Special case for getting the replication region for
      // TPUCompilationResultsOp.
      return op->getAttrOfType<StringAttr>(kTpuCompilationStatus);
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 39.4K bytes
    - Viewed (0)
  9. tensorflow/compiler/mlir/tensorflow/transforms/cluster_ops_by_policy.cc

        Operation *op, const ClusteringPolicySet &policies,
        const std::function<bool(Operation *op)> &filter) {
      // Check that op has no side effects. This guarantees that we will not
      // reorder side-effecting ops during cluster formation.
      if (!isMemoryEffectFree(op)) return std::nullopt;
    
      // Operation rejected by the custom filter.
      if (filter && !filter(op)) return std::nullopt;
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 27.9K bytes
    - Viewed (0)
  10. tensorflow/compiler/mlir/lite/transforms/lower_static_tensor_list.cc

        mlir::Operation *op = use.getOwner();
        if (llvm::isa<TF::TensorListGetItemOp>(op) ||
            llvm::isa<TF::TensorListLengthOp>(op) ||
            llvm::isa<TF::TensorListPushBackOp>(op) ||
            llvm::isa<TF::TensorListReserveOp>(op) ||
            llvm::isa<TF::TensorListSetItemOp>(op) ||
            llvm::isa<TF::TensorListStackOp>(op) ||
            llvm::isa<TF::TensorListResizeOp>(op)) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Jun 11 20:00:43 UTC 2024
    - 70.7K bytes
    - Viewed (0)
Back to top