- Sort Score
- Result 10 results
- Languages All
Results 121 - 130 of 263 for created (0.19 sec)
-
tensorflow/c/kernels.cc
class KernelBuilderFactory : public ::tensorflow::kernel_factory::OpKernelFactory { public: explicit KernelBuilderFactory(TF_KernelBuilder* builder) : builder_(builder) {} ::tensorflow::OpKernel* Create( ::tensorflow::OpKernelConstruction* context) override { if (builder_->compute_function) return new ::tensorflow::COpKernel(context, builder_->create_function,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 28 22:53:47 UTC 2024 - 36K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/passes/quantization_patterns.cc
// Add i32 -> i8 requantization. UniformQuantizeOp uniform_quant_op = rewriter.create<UniformQuantizeOp>( op.getLoc(), func_result_type, op.getResults()); cast<func::ReturnOp>(entry_func_op.getBody().front().getTerminator()) .setOperand(0, uniform_quant_op); } template <typename GemmStyleOp> // Creates a quantized bias pattern for static and dynamic shape case
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 03 06:04:36 UTC 2024 - 41.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/sparsecore/embedding_pipelining.cc
func::FuncOp parent_func, ModuleOp module, const std::string& name, bool flag_for_inlining) { // Moves all of the Operations in 'ops' into a newly created func.FuncOp // function named 'name' and replaces the original ops with a call to the // newly created function using a tf.StatefulPartitionedCall. Here, // 'parent_func' is the function that holds the original set of ops.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 92.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/ir/tf_ops_n_z.cc
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 09 22:07:10 UTC 2024 - 170.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/executor_tpuv1_island_coarsening.cc
ModuleOp module) { SymbolTableCollection table; SymbolUserMap symbol_map(table, module); llvm::DenseMap<func::FuncOp, llvm::DenseSet<func::FuncOp>> caller_callee_map; // Creates work queue for determining reachability below. std::queue<func::FuncOp> function_worklist; for (auto func : module.getOps<func::FuncOp>()) { for (auto user : symbol_map.getUsers(func)) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 17 17:58:54 UTC 2024 - 27.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/debugging/dump_tensor_op.cc
} // `DumpTensor` op saves entire value of input to as a tensor proto into a // specified directory and filename. When enabled is set to false, op is // disabled and won't save any value. It also creates `QuantizationUnit` proto // with `func_name` and `node_name` to identify the op. REGISTER_OP("DumpTensor") .Input("tensor_data: T") .Attr("log_dir_path: string") .Attr("file_name: string") .Attr("T: type")
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Feb 22 03:12:17 UTC 2024 - 4.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/convert_tpu_model_to_cpu.cc
module_op.emitError() << "quant-convert-tpu-model-to-cpu pattern " "conversion did not converge."; signalPassFailure(); return; } } } // namespace // Creates an instance of `ConvertTpuModelToCpuPass`. std::unique_ptr<OperationPass<ModuleOp>> CreateConvertTpuModelToCpuPass() { return std::make_unique<ConvertTpuModelToCpuPass>(); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 5.5K bytes - Viewed (0) -
tensorflow/compiler/jit/tests/device_compiler_test_helper.cc
#include "tensorflow/compiler/jit/xla_compilation_cache.pb.h" #include "xla/service/hlo.pb.h" #include "tensorflow/core/platform/path.h" #include "tensorflow/core/public/session.h" namespace tensorflow { namespace { // Creates a float tensor of linearly increasing values, starting from offset. Tensor CreateInputTensor(const TensorShape& shape, float offset) { Tensor tensor(DT_FLOAT, shape); for (int64 i = 0; i < tensor.flat<float>().size(); ++i) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Feb 09 08:24:16 UTC 2024 - 6.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tpu_space_to_depth_pass.cc
auto reshape_sizes = DenseIntElementsAttr::get(reshape_type, new_shape); auto reshape_value = builder->create<TF::ConstOp>(input.getLoc(), reshape_sizes); return builder->create<TF::ReshapeOp>(input.getLoc(), reshape_result_type, input, reshape_value); } // Creates transpose op for shape to depth transform. TF::TransposeOp GetTransposeOpForConv2DFilter(OpBuilder* builder, Value input) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 29.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/passes/prepare_quantize.cc
ctx); if (failed(applyPatternsAndFoldGreedily(func_op, std::move(patterns_2)))) { signalPassFailure(); } } } } // namespace // Creates an instance of the TensorFlow dialect PrepareQuantize pass. std::unique_ptr<OperationPass<ModuleOp>> CreatePrepareQuantizePass( const bool enable_per_channel_quantized_weight, const int bit_width) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Apr 03 05:11:03 UTC 2024 - 8.1K bytes - Viewed (0)