Search Options

Results per page
Sort
Preferred Languages
Advance

Results 41 - 50 of 56 for tensorlist (0.26 sec)

  1. tensorflow/compiler/mlir/tensorflow/ir/tf_ops_n_z.cc

    namespace {
    // If the input of TensorListGetItemOp is TensorListFromTensorOp and the
    // TensorListFromTensorOp is only used by TensorListGetItemOp (not modified by
    // other TensorList ops), we can convert it to a GatherOp.
    class ConvertTensorListGetItemOpOfTensorListFromTensorOpToGather
        : public OpRewritePattern<TensorListGetItemOp> {
      using OpRewritePattern<TensorListGetItemOp>::OpRewritePattern;
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu May 09 22:07:10 UTC 2024
    - 170.8K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/tfr/python/tfr_gen.py

              ssa_value, _ = value
              self.symbol_table.insert_symbol(key, ssa_value, t)
          elif len(values) == 1:
            name, tys = values[0]
            if ty == TFRTypes.TENSOR_LIST:
              # assign single tensor_list to multiple variables
              for idx, key in enumerate(targets):
                idx_name = self._ssa_name('idx')
                self._emit_with_loc(
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Apr 27 15:27:03 UTC 2022
    - 55.8K bytes
    - Viewed (0)
  3. src/cmd/go/internal/modload/vendor.go

    				// is in the build list and is the selected version of its path.
    				// If this information is new, record it.
    				if v, ok := vendorVersion[mod.Path]; !ok || gover.ModCompare(mod.Path, v, mod.Version) < 0 {
    					vendorList = append(vendorList, mod)
    					vendorVersion[mod.Path] = mod.Version
    				}
    			}
    		}
    	})
    }
    
    // checkVendorConsistency verifies that the vendor/modules.txt file matches (if
    Registered: Wed Jun 12 16:32:35 UTC 2024
    - Last Modified: Mon Jul 24 18:09:22 UTC 2023
    - 9.5K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/tfr/passes/decompose.cc

        // Create the new operands. This is mapping the operands from the target
        // TF ops to the TFR function arguments. If the TFR function argument is
        // a tensor_list, a "tfr.build_list" op is used to concat the available
        // TF op operands. If the TFR function argument isn't a tensor/tensor_list,
        // a constant is created by using the attribute stored in the TF op or the
        // default value in the argument attribute.
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 14.6K bytes
    - Viewed (0)
  5. tensorflow/compiler/mlir/tensorflow/translate/import_model.cc

        shape_inference::InferenceContext* shape_context =
            shape_refiner_->GetContext(&node);
        return shape_ic(shape_context);
      }
    
      // Treat TensorList init ops specially here as the op requires knowing its
      // element dtype.
      // TODO(jpienaar): Reconsider post refactoring shape functions.
      if (node.type_string() == "TensorListReserve" ||
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed May 01 11:17:36 UTC 2024
    - 183.2K bytes
    - Viewed (0)
  6. tensorflow/compiler/mlir/tfr/ir/tfr_types.h

    };
    
    class TFRTensorListType : public detail::TFRTypeImpl<TFRTensorListType> {
     public:
      using TFRBase::TFRBase;
      static constexpr StringLiteral name = "tfr.tensor_list";
      static std::string getTypeName() { return "TFRTensorListType"; }
    };
    
    class TFRAttrType : public Type::TypeBase<TFRAttrType, TFRType, TypeStorage> {
     public:
      using Base::Base;
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Dec 05 07:17:01 UTC 2023
    - 4.1K bytes
    - Viewed (0)
  7. tensorflow/compiler/mlir/tfr/passes/raise_to_tf.cc

          SmallVector<Value, 4> tensor_list;
          for (int i = res.index(); i < new_op->getNumResults(); i++) {
            Value new_res = new_op->getResult(i);
            auto casted =
                rewriter.create<CastOp>(loc, unconstrainted_type, new_res);
            tensor_list.push_back(casted.getOut());
          }
          auto list_op = rewriter.create<BuildListOp>(loc, res_type, tensor_list);
          new_results.push_back(list_op.getOut());
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 21.8K bytes
    - Viewed (0)
  8. tensorflow/compiler/mlir/tfrt/tests/mlrt/while_to_map_fn.mlir

      // CHECK: [[tensor_list:%.*]] = "tf.TensorListReserve"([[shape:%.*]], [[reserve_size:%.*]]) {
      %0 = "tf.TensorListReserve"(%cst_1, %cst_2) {device = "/job:localhost/replica:0/task:0/device:CPU:0"} : (tensor<i32>, tensor<i32>) -> tensor<!tf_type.variant<tensor<*xf32>>>
      // CHECK: [[map_fn_result:%.*]] = tf_mlrt.tf_map_fn([[reserve_size]], [[tensor_list]], %arg0)
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Apr 23 06:40:22 UTC 2024
    - 68.6K bytes
    - Viewed (0)
  9. tensorflow/compiler/mlir/quantization/tensorflow/python/py_function_lib.py

    ) -> Mapping[str, core.Tensor]:
      """Converts TensorLike values of `sample` to Tensors.
    
      Creates a copy of `sample`, where each value is converted to Tensors
      unless it is already a Tensor.
      The values are not converted in-place (i.e. `sample` is not mutated).
    
      Args:
        sample: A representative sample, which is a map of {name -> tensorlike
          value}.
    
      Returns:
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri May 31 05:32:11 UTC 2024
    - 27.4K bytes
    - Viewed (0)
  10. tensorflow/c/eager/tape.h

        const int64_t id = target_tensor_ids[i];
        if (output_gradients.empty() || output_gradients[i] == nullptr) {
          auto tensor_it = tensor_tape.find(id);
          if (tensor_it != tensor_tape.end() && tensor_it->second != -1) {
            auto op_it = op_tape.find(tensor_it->second);
            if (op_it == op_tape.end()) {
              return errors::Internal(
                  "Internal state of the gradient tape is invalid: "
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Apr 02 12:40:29 UTC 2024
    - 47.2K bytes
    - Viewed (0)
Back to top