Search Options

Results per page
Sort
Preferred Languages
Advance

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

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

    bool IsDynamicPadderOp(const TypeID& type_id);
    
    // Returns True if this op has a Tf2XLA fallback. Currently, this is not the
    // inverse of the !IsOpLegalizedWithMlir, but it should be.
    bool HasTf2XlaFallback(const TypeID& type_id);
    
    // Whether this type is allowed to have a TF2XLA fallback.
    bool IsOpAllowedTf2xlaFallback(const TypeID& type_id);
    
    // Whether this type is Preferred to use a TF2XLA fallback kernel when using
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon Feb 26 20:53:25 UTC 2024
    - 2K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/lite/experimental/tac/hardwares/target_hardware.h

        // TODO(b/177376459): remove this.
        internal::RegisterTargetHardwareOp(mlir::TypeID::get<Hardware>(),
                                           mlir::TypeID::get<Op>(),
                                           target_hardware_op_factory);
        internal::RegisterTargetHardwareOpFactory(mlir::TypeID::get<Hardware>(),
                                                  mlir::TypeID::get<Op>(),
                                                  target_hardware_op_factory);
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue May 09 21:39:59 UTC 2023
    - 7.3K 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. tensorflow/compiler/mlir/tensorflow/ir/tpu_embedding_ops_registry.h

      template <typename OpType>
      void Add() {
        ops_type_ids_.insert(TypeID::get<OpType>());
      }
    
      // Returns the type id of the ops in the TPUEmbeddingOpRegistry.
      const llvm::SmallDenseSet<mlir::TypeID>& GetOpsTypeIds();
    
      // Returns the global registry.
      static TPUEmbeddingOpsRegistry& Global();
    
     private:
      llvm::SmallDenseSet<mlir::TypeID> ops_type_ids_{};
    };
    }  // namespace TF
    }  // namespace mlir
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Dec 14 05:42:28 UTC 2022
    - 2K bytes
    - Viewed (0)
  5. src/encoding/gob/decoder.go

    	}
    	dec.r = r
    	dec.wireType = make(map[typeId]*wireType)
    	dec.decoderCache = make(map[reflect.Type]map[typeId]**decEngine)
    	dec.ignorerCache = make(map[typeId]**decEngine)
    	dec.countBuf = make([]byte, 9) // counts may be uint64s (unlikely!), require 9 bytes
    
    	return dec
    }
    
    // recvType loads the definition of a type.
    func (dec *Decoder) recvType(id typeId) {
    	// Have we already seen this type? That's an error
    Registered: Wed Jun 12 16:32:35 UTC 2024
    - Last Modified: Fri Sep 08 19:04:28 UTC 2023
    - 7.2K bytes
    - Viewed (0)
  6. 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)
  7. tensorflow/compiler/mlir/lite/experimental/tac/hardwares/gpu_hardware.h

    class GpuHardware : public TargetHardware {
     public:
      static constexpr char kId[] = "GPU";
      mlir::RewritePatternSet GetTransformations(
          MLIRContext* context) const override;
    
      mlir::TypeID GetTypeId() const override {
        return mlir::TypeID::get<GpuHardware>();
      }
    
      double GetHardwareSwitchingCost(const TargetHardware* from,
                                      size_t buffer_size) const override;
    };
    }  // namespace tac
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Jan 27 15:05:02 UTC 2022
    - 1.7K bytes
    - Viewed (0)
  8. tensorflow/compiler/mlir/lite/experimental/tac/examples/example_hardware.h

     public:
      static constexpr char kId[] = "ExampleHardware";
    
      mlir::RewritePatternSet GetTransformations(
          MLIRContext* context) const override;
    
      mlir::TypeID GetTypeId() const override {
        return mlir::TypeID::get<ExampleHardware>();
      }
    
      bool IsNotSupportedOp(mlir::Operation* op) const override { return false; }
    
      float AdvantageOverCPU() const override { return 5.0; }
    };
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Jan 27 15:05:02 UTC 2022
    - 1.5K bytes
    - Viewed (0)
  9. tensorflow/compiler/mlir/lite/experimental/tac/hardwares/nnapi_hardware.h

     public:
      static constexpr char kId[] = "NNAPI";
    
      mlir::RewritePatternSet GetTransformations(
          MLIRContext* context) const override;
    
      mlir::TypeID GetTypeId() const override {
        return mlir::TypeID::get<NNAPIHardware>();
      }
    
      bool IsNotSupportedOp(mlir::Operation* op) const override { return false; }
    
      float AdvantageOverCPU() const override { return 5.0; }
    };
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Jan 27 15:05:02 UTC 2022
    - 1.5K bytes
    - Viewed (0)
  10. tensorflow/compiler/mlir/tensorflow/ir/tf_dialect.h

      // A hook may use the public addOperations() method to add additional
      // operations to the dialect. Hooks will only apply to subsequent
      // instantations of the Dialect/MLIRContext.
      static void RegisterAdditionalOperationHook(TypeID uniqueId,
                                                  AdditionalOpFunction fn);
    
      // Re-define publicly the protected addOperations() method from the Dialect
      // class, usually used in a Dialect constructor. This allows hook
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Jun 08 01:19:25 UTC 2023
    - 4.8K bytes
    - Viewed (0)
Back to top