Search Options

Results per page
Sort
Preferred Languages
Advance

Results 131 - 140 of 234 for created (0.08 sec)

  1. tensorflow/compiler/mlir/lite/transforms/modify_io_nodes.cc

        return;
      }
    
      auto new_func_type =
          builder.getFunctionType(new_input_types, new_output_types);
      func.setType(new_func_type);
    }
    }  // namespace
    
    // Creates an instance of the TensorFlow Lite modify io nodes pass.
    std::unique_ptr<OperationPass<func::FuncOp>> CreateModifyIONodesPass(
        Type input_type, Type output_type) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 8.9K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/tensorflow/transforms/xla_call_module_serialization.cc

        return WalkResult::advance();
      });
      if (result.wasInterrupted()) {
        return failure();
      }
    
      return builder.getArrayAttr(function_list);
    }
    
    // Creates a pruned module containing the XlaCallModule's entry function and
    // other functions transitively called by the entry function.
    FailureOr<OwningOpRef<ModuleOp>> PruneStablehloModule(
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 9.5K bytes
    - Viewed (0)
  3. tensorflow/cc/saved_model/fingerprinting.cc

            result_hash, tensorflow::Fingerprint64(concrete_function_serialized));
      }
      // TODO(b/241294832): Complete canonicalization of `object_graph_def.nodes`.
      return result_hash;
    }
    
    // Creates a FingerprintDef proto from a SavedModel and the checkpoint meta file
    // (.index) in `export_dir`.
    absl::StatusOr<FingerprintDef> CreateFingerprintDefPb(
        absl::string_view export_dir, std::string pb_file) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Mar 26 00:24:44 UTC 2024
    - 9.8K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/lite/transforms/default_quant_params.cc

      } else {
        builder.setInsertionPointToStart(&block);
      }
      TypeAttr type_attr = TypeAttr::get(new_type);
      auto quantize = builder.create<TFL::QuantizeOp>(value.getLoc(), new_type,
                                                      value, type_attr);
      auto dequantize = builder.create<TFL::DequantizeOp>(
          value.getLoc(), expressed_type, quantize.getOutput());
      value.replaceAllUsesWith(dequantize);
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 9.4K bytes
    - Viewed (0)
  5. tensorflow/compiler/mlir/tensorflow/translate/export_graphdef.cc

      nodes_[inst] = node;
      return absl::OkStatus();
    }
    
    bool IsEntryFunctionArg(BlockArgument arg) {
      return arg.getParentRegion()->getParentOfType<FuncOp>().getName() == "main";
    }
    
    // Creates argument nodes from Block argument. If a name is supplied, that
    // name will be used instead of generating a unique name.
    Status Exporter::AddArgumentNode(BlockArgument arg, unsigned index,
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed May 01 11:17:36 UTC 2024
    - 35.2K bytes
    - Viewed (0)
  6. tensorflow/compiler/mlir/tf2xla/api/v2/tf_executor_to_graph.cc

      nodes_[inst] = node;
      return absl::OkStatus();
    }
    
    bool IsEntryFunctionArg(BlockArgument arg) {
      return arg.getParentRegion()->getParentOfType<FuncOp>().getName() == "main";
    }
    
    // Creates argument nodes from Block argument. If a name is supplied, that
    // name will be used instead of generating a unique name.
    Status Exporter::AddArgumentNode(BlockArgument arg, unsigned index,
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Jun 06 23:04:51 UTC 2024
    - 35.2K bytes
    - Viewed (0)
  7. tensorflow/compiler/mlir/quantization/tensorflow/passes/prepare_quantize.cc

                /*quant_dim=*/-1, num_bits, narrow_range, is_signed);
            builder.setInsertionPoint(block, insertion_point);
            auto q_op = builder.create<quantfork::QuantizeCastOp>(
                loc, params.getValue(), arg);
            auto dq_op = builder.create<quantfork::DequantizeCastOp>(
                loc, input_type, q_op.getResult());
            arg.replaceAllUsesWith(dq_op.getResult());
            q_op.setOperand(arg);
          }
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 17.2K bytes
    - Viewed (0)
  8. tensorflow/compiler/mlir/tensorflow/analysis/resource_alias_analysis.cc

            // Keep track of IDs of resources that are allocated by ops with
            // `UniqueResourceAllocation` trait, this can be utilized for while-loop
            // parallelization (every iteration creates a new unique resource).
            if (op->hasTrait<OpTrait::TF::UniqueResourceAllocation>()) {
              unique_resource_allocation_ids_.insert(resource_handle.id);
            }
          }
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed May 15 09:04:13 UTC 2024
    - 28.2K bytes
    - Viewed (0)
  9. tensorflow/compiler/mlir/tfr/passes/raise_to_tf.cc

        const llvm::StringMap<Attribute>& derived_attrs) const {
      // Create the new op
      Location loc = call_op.getLoc();
      rewriter.setInsertionPointAfter(call_op);
      std::string tf_op_name = GetTFOpName(call_op.getCallee());
      OperationState new_state(loc, tf_op_name, inputs, output_types, attr_list);
      Operation* new_op = rewriter.create(new_state);
      if (materialize_derived_attrs_) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 21.8K bytes
    - Viewed (0)
  10. tensorflow/compiler/mlir/tensorflow/transforms/sparsecore/embedding_program_key.cc

      auto launch = builder->create<tf_device::LaunchOp>(
          before_op->getLoc(), builder->getStringAttr(host_device),
          launch_result_types);
      launch.getBody().push_back(launch_block);
    
      builder->setInsertionPointToEnd(&launch.GetBody());
      builder->create<tf_device::ReturnOp>(before_op->getLoc(), launch_results);
    
      return launch;
    }
    
    // Creates a new_launch after `before_op` with `ops_to_move` from
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 17.4K bytes
    - Viewed (0)
Back to top