Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 10 of 230 for TypeId (0.15 sec)

  1. tensorflow/compiler/mlir/tf2xla/transforms/legalization_op_config.cc

          TypeID::get<TF::CosOp>(),
          TypeID::get<TF::TanOp>(),
          TypeID::get<TF::DiagPartOp>(),
          TypeID::get<TF::EinsumOp>(),
          TypeID::get<TF::ExpOp>(),
          TypeID::get<TF::Expm1Op>(),
          TypeID::get<TF::FakeQuantWithMinMaxArgsOp>(),
          TypeID::get<TF::FloorOp>(),
          TypeID::get<TF::IFFTOp>(),
          TypeID::get<TF::ImagOp>(),
          TypeID::get<TF::IsFiniteOp>(),
          TypeID::get<TF::IsInfOp>(),
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Apr 24 04:08:35 UTC 2024
    - 21.7K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/tensorflow/transforms/tpu_validate_inputs.cc

      static auto* ops = [] {
        llvm::SmallDenseSet<mlir::TypeID, 32>* ops_set =
            new llvm::SmallDenseSet<mlir::TypeID, 32>{
                TypeID::get<mlir::ModuleOp>(),
                TypeID::get<mlir::tf_executor::GraphOp>(),
                TypeID::get<mlir::func::ReturnOp>(),
                TypeID::get<mlir::func::FuncOp>(),
                TypeID::get<mlir::tf_executor::YieldOp>(),
                TypeID::get<mlir::tf_executor::IslandOp>(),
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue May 07 06:51:01 UTC 2024
    - 21.5K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/tf2xla/transforms/legalization_op_config_test.cc

      EXPECT_TRUE(IsTypeLegalizedWithMlir(TypeID::get<TF::ModOp>()));
      EXPECT_FALSE(HasTf2XlaFallback(TypeID::get<TF::ModOp>()));
      EXPECT_FALSE(IsOpAllowedTf2xlaFallback(TypeID::get<TF::ModOp>()));
      EXPECT_FALSE(IsOpAllowedTf2xlaPreferred(TypeID::get<TF::ModOp>()));
    }
    
    TEST_F(LegalizationOpConfigTest, ExpectsTrueForTF2XLATypeID) {
      EXPECT_TRUE(HasTf2XlaFallback(TypeID::get<TF::AllOp>()));
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu May 30 03:31:01 UTC 2024
    - 8.4K bytes
    - Viewed (0)
  4. src/encoding/gob/type.go

    	}
    	return ut
    }
    
    // A typeId represents a gob Type as an integer that can be passed on the wire.
    // Internally, typeIds are used as keys to a map to recover the underlying type info.
    type typeId int32
    
    var typeLock sync.Mutex // set while building a type
    const firstUserId = 64  // lowest id number granted to user
    
    type gobType interface {
    	id() typeId
    	setId(id typeId)
    	name() string
    Registered: Wed Jun 12 16:32:35 UTC 2024
    - Last Modified: Thu May 16 02:00:26 UTC 2024
    - 27.2K bytes
    - Viewed (0)
  5. tensorflow/compiler/mlir/tensorflow/transforms/shape_inference.h

    #include "mlir/IR/Types.h"  // from @llvm-project
    #include "mlir/Support/LLVM.h"  // from @llvm-project
    #include "mlir/Support/LogicalResult.h"  // from @llvm-project
    #include "mlir/Support/TypeID.h"  // from @llvm-project
    
    namespace mlir {
    namespace TF {
    
    // Returns whether type can be further refined.
    bool CanBeRefined(Type type);
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Apr 24 12:49:45 UTC 2024
    - 3.5K bytes
    - Viewed (0)
  6. tensorflow/compiler/mlir/tensorflow/ir/tf_ops.cc

    // These are invoked at construction time.
    static DenseMap<TypeID, TensorFlowDialect::AdditionalOpFunction>
        &GetAdditionalOperationHooks() {
      static auto *additional_operation_hooks =
          new DenseMap<TypeID, TensorFlowDialect::AdditionalOpFunction>();
      return *additional_operation_hooks;
    }
    
    void TensorFlowDialect::RegisterAdditionalOperationHook(
        TypeID id, AdditionalOpFunction fn) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 14.6K bytes
    - Viewed (0)
  7. tensorflow/compiler/mlir/tf2xla/transforms/verify_tfxla_legalization.cc

        "such as a random number generator.";
    
    // TODO(b/282188914) remove the operations to skip once tests are fixed.
    static const DenseSet<mlir::TypeID>* operations_to_skip =
        new DenseSet<mlir::TypeID>{mlir::TypeID::get<mhlo::EinsumOp>()};
    
    class VerifyTFXLALegalization
        : public impl::VerifyTFXLALegalizationBase<VerifyTFXLALegalization> {
     public:
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 6.1K bytes
    - Viewed (0)
  8. tensorflow/c/kernels_experimental.cc

            "The output variant hasn't been initialized");
      }
    
      if (a.TypeId() != b.TypeId()) {
        return ::tensorflow::errors::Internal(
            "BinaryOpVariants: Variants a and b have different "
            "type ids.  Type names: '",
            a.TypeName(), "' vs. '", b.TypeName(), "'");
      }
    
      if (a.TypeId() == tensorflow::TypeIndex::Make<::tensorflow::TensorList>()) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Apr 23 06:12:29 UTC 2024
    - 30.9K bytes
    - Viewed (0)
  9. tensorflow/compiler/mlir/tfrt/ir/tfrt_fallback.cc

    #include "mlir/Support/LLVM.h"  // from @llvm-project
    
    namespace tfrt {
    namespace fallback {
    
    FallbackDialect::FallbackDialect(MLIRContext *context)
        : Dialect(/*name=*/"tfrt_fallback", context,
                  TypeID::get<FallbackDialect>()) {
      addTypes<TFTensorType, TFAllocatorType>();
    
      addOperations<
    #define GET_OP_LIST
    #include "tensorflow/compiler/mlir/tfrt/ir/tfrt_fallback.cpp.inc"
          >();
    }
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 2.4K bytes
    - Viewed (0)
  10. tensorflow/compiler/mlir/tfrt/ir/tfrt_fallback_sync.cc

    namespace tfrt {
    namespace fallback_sync {
    
    FallbackSyncDialect::FallbackSyncDialect(MLIRContext *context)
        : Dialect(/*name=*/"tfrt_fallback_sync", context,
                  TypeID::get<FallbackSyncDialect>()) {
      context->getOrLoadDialect<tfrt::fallback::FallbackDialect>();
      context->getOrLoadDialect<compiler::TFRTDialect>();
      context->getOrLoadDialect<corert::CoreRTDialect>();
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu May 16 01:32:39 UTC 2024
    - 2.7K bytes
    - Viewed (0)
Back to top