Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 10 of 34 for new_types (0.2 sec)

  1. tensorflow/compiler/mlir/tensorflow/transforms/tfg-to-tfe.cc

        llvm::SmallVector<Type, 2> new_types(op->getResultTypes());
        new_types.back() = rewriter.getType<tf_executor::ControlType>();
    
        llvm::SmallVector<Value, 2> new_operands;
        FilterOutBlockArgControlDep(operands, new_operands);
    
        rewriter.replaceOpWithNewOp<tf_executor::ControlTriggerOp>(
            op, new_types, new_operands, op->getAttrs());
        return success();
      }
    };
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri May 17 17:58:54 UTC 2024
    - 21.2K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/lite/transforms/while_loop_outline.cc

      SmallVector<Type, 4> new_types;
      new_types.reserve(operands_size);
      new_types.append(while_op.getResultTypes().begin(),
                       while_op.getResultTypes().end());
      for (auto extra_operand : extra_operands)
        new_types.push_back(extra_operand.getType());
    
      auto new_while_op = OpBuilder(while_op).create<WhileOp>(
          while_op.getLoc(), new_types, operands, while_op->getAttrs());
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 11.8K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/quantization/tensorflow/passes/insert_main_function.cc

        arg_idx += func_op.getNumArguments();
        llvm::ArrayRef<Type> new_types = llvm::ArrayRef(
            result_types.begin() + result_idx, func_op.getNumResults());
        result_idx += func_op.getNumResults();
    
        auto call_op = builder.create<TF::PartitionedCallOp>(
            module_op.getLoc(), new_types, new_args,
            SymbolRefAttr::get(context, func_op.getSymName()),
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 16.5K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/tensorflow/transforms/resource_op_lifting.cc

        int64_t index = entry.getFirst();
        Type new_type = entry.getSecond().first;
        int64_t updated_index = entry.getSecond().second;
        auto operand = caller->getOperand(index);
        builder.setInsertionPoint(caller);
        new_operands[index] = builder.create<TF::ReadVariableOp>(
            caller->getLoc(), ArrayRef<Type>{new_type}, ArrayRef<Value>{operand});
        caller->setOperand(index, new_operands[index]);
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 55.1K bytes
    - Viewed (0)
  5. tensorflow/compiler/mlir/tensorflow/transforms/decompose_optionals.cc

        Type newType = nullptr;
        if (auto variant = dyn_cast<TF::VariantType>(elementType)) {
          ArrayRef<TensorType> sub = variant.getSubtypes();
          if (sub.size() == 1) {
            auto inner = sub[0];
            if (!isa<TF::VariantType>(inner)) {
              newType = inner;
            }
          }
        }
        if (!newType) {
          return failure();
        }
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Jun 12 21:18:05 UTC 2024
    - 9.8K bytes
    - Viewed (0)
  6. tensorflow/compiler/mlir/lite/stablehlo/transforms/legalize_hlo_conversions/util.cc

            shape[spatial_dim_start + i];
      }
      auto new_type =
          RankedTensorType::get(transposed_shape, type.getElementType());
      auto permutation = DenseIntElementsAttr::get(
          RankedTensorType::get({type.getRank()}, rewriter.getI64Type()),
          permutation_array);
      return {new_type, permutation};
    }
    
    Value InsertTranspose(Value value, int batch_dim, int feature_dim,
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 10.1K bytes
    - Viewed (0)
  7. tensorflow/compiler/mlir/quantization/tensorflow/passes/prepare_lifting.cc

      // Multiply the quantization parameters by the multiplier.
      QuantizedType new_qtype;
      auto element_type = mlir::cast<TensorType>(q_op.getType()).getElementType();
      if (auto uniform_type = llvm::dyn_cast<UniformQuantizedType>(element_type)) {
        if (multiplier_attr.isSplat()) {
          double new_scale = multiplier_array.front() * uniform_type.getScale();
          new_qtype = UniformQuantizedType::get(
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri May 17 17:58:54 UTC 2024
    - 13.3K bytes
    - Viewed (0)
  8. tensorflow/compiler/mlir/lite/transforms/default_quant_params.cc

                                               quant::QuantParams quant_params) {
      Type expressed_type = value.getType();
      Type new_type = quant_params.castFromExpressedType(expressed_type);
      // This value isn't an expressed type (float), skip.
      if (!new_type) return;
    
      Block &block = value.getParentRegion()->front();
      Operation *op = value.getDefiningOp();
      if (op) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 9.4K bytes
    - Viewed (0)
  9. tensorflow/compiler/mlir/quantization/common/quantization_lib/quantization_utils.cc

            new_scales, qtype.getZeroPoints(), qtype.getQuantizedDimension(),
            qtype.getStorageTypeMin(), qtype.getStorageTypeMax());
        if (const auto new_type = new_ele_type.castFromExpressedType(
                quant::QuantizedType::castToExpressedType(input))) {
          return TypeAttr::get(new_type);
        }
      }
      // Currently, we only support per-axis quantized type.
      return {};
    }
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed May 08 02:10:16 UTC 2024
    - 43.2K bytes
    - Viewed (0)
  10. platforms/core-configuration/configuration-cache/src/main/kotlin/org/gradle/internal/cc/impl/serialize/DefaultClassDecoder.kt

                } else {
                    scope.exportClassLoader
                }
            } else {
                this.classLoader
            }
            val newType = Class.forName(name, false, classLoader)
            classes.putInstance(id, newType)
            return newType
        }
    
        private
        fun ReadContext.readScope(): ClassLoaderScope {
            val id = readSmallInt()
            val scope = scopes.getInstance(id)
    Registered: Wed Jun 12 18:38:38 UTC 2024
    - Last Modified: Sat Jun 08 11:29:30 UTC 2024
    - 3.2K bytes
    - Viewed (0)
Back to top