Search Options

Results per page
Sort
Preferred Languages
Advance

Results 11 - 20 of 2,529 for dtype (0.16 sec)

  1. tensorflow/c/c_api_experimental.cc

    }
    
    TFE_TensorHandle* TFE_NewTensorHandleFromScalar(TF_DataType data_type,
                                                    void* data, size_t len,
                                                    TF_Status* status) {
      auto dtype = static_cast<tensorflow::DataType>(data_type);
      DCHECK(tensorflow::DataTypeCanUseMemcpy(dtype));
    
      tensorflow::Tensor tensor(dtype, tensorflow::TensorShape({}));
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Mon Apr 15 03:35:10 GMT 2024
    - 29.4K bytes
    - Viewed (0)
  2. tensorflow/c/eager/c_api_test.cc

                name_and_attrs.attr().find("dtype")->second.type());
      TF_DeleteBuffer(serialized_attr_values);
    
      TFE_Op* var_op_2 = TFE_NewOp(ctx, "VarHandleOp", status);
    
      string serialized_dtype;
      ASSERT_TRUE(name_and_attrs.attr().find("dtype")->second.SerializeToString(
          &serialized_dtype));
      TFE_OpSetAttrValueProto(
          var_op_2, "dtype",
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Thu Aug 03 20:50:20 GMT 2023
    - 94.6K bytes
    - Viewed (1)
  3. tensorflow/c/eager/custom_device_test.cc

          TFE_NewOp(context.get(), "VarHandleOp", status.get()), TFE_DeleteOp);
      ASSERT_TRUE(TF_GetCode(status.get()) == TF_OK) << TF_Message(status.get());
      TFE_OpSetAttrType(op.get(), "dtype", TF_FLOAT);
      TFE_OpSetAttrShape(op.get(), "shape", {}, 0, status.get());
      TFE_OpSetAttrString(op.get(), "container", "", 0);
      TFE_OpSetAttrString(op.get(), "shared_name", "", 0);
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Thu Aug 27 23:39:24 GMT 2020
    - 18.4K bytes
    - Viewed (0)
  4. tensorflow/c/experimental/gradients/math_grad.cc

      auto dtype = input->DataType();
      if (DataTypeIsFloating(BaseType(dtype)) ||
          DataTypeIsInteger(BaseType(dtype))) {
        return tensorflow::ops::Identity(ctx, input, output, name);
      } else if (!DataTypeIsComplex(BaseType(dtype)) &&
                 BaseType(dtype) != DT_VARIANT) {
        return errors::InvalidArgument(
            "Expected numeric or variant tensor, got dtype ", dtype);
      }
    C++
    - Registered: Tue Mar 26 12:39:09 GMT 2024
    - Last Modified: Wed Feb 28 13:53:47 GMT 2024
    - 15.2K bytes
    - Viewed (0)
  5. tensorflow/c/eager/tape.h

        call_state_.top().backward_tape->RecordOperation(
            op_type, output_tensors, input_tensor_id, input_dtypes,
            backward_function_getter, backward_function_deleter);
        return absl::OkStatus();
      }
      if (!ShouldRecord(input_tensor_id, input_dtypes)) {
        return absl::OkStatus();
      }
    
      // We may need to allocate zero inputs for trainable dtypes we don't have JVPs
      // for. Make sure they get cleaned up.
    C
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Tue Apr 02 12:40:29 GMT 2024
    - 47.2K bytes
    - Viewed (1)
  6. tensorflow/c/eager/parallel_device/parallel_device_lib_test.cc

          TFE_NewOp(context.get(), "VarHandleOp", status.get()), TFE_DeleteOp);
      ASSERT_TRUE(TF_GetCode(status.get()) == TF_OK) << TF_Message(status.get());
      TFE_OpSetAttrType(handle_op.get(), "dtype", TF_FLOAT);
      TFE_OpSetAttrShape(handle_op.get(), "shape", /*dims=*/nullptr, /*num_dims=*/0,
                         status.get());
      ASSERT_TRUE(TF_GetCode(status.get()) == TF_OK) << TF_Message(status.get());
      auto outputs =
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Thu Jul 08 23:47:35 GMT 2021
    - 15.3K bytes
    - Viewed (0)
  7. tensorflow/c/eager/immediate_execution_tensor_handle.cc

      const char* device_name = DeviceName(&s);
      if (!s.ok()) {
        device_name = "<error fetching device name>";
      }
      return absl::StrCat("TensorHandle(", value_string, ", shape=", shape_string,
                          ", dtype=", DataType_Name(DataType()), ", device=\"",
                          device_name, "\")");
    }
    
    Status ImmediateExecutionTensorHandle::SummarizeValue(
        std::string& summary) const {
      Status status;
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Thu Feb 15 09:49:45 GMT 2024
    - 2.1K bytes
    - Viewed (0)
  8. tensorflow/c/eager/abstract_tensor_handle.cc

        shape_string = "<error computing shape>";
      } else {
        shape_string = shape.DebugString();
      }
      return absl::StrCat("TensorHandle(shape=", shape_string,
                          ", dtype=", DataType_Name(DataType()),
                          ", type=", FullType().DebugString(), ")");
    }
    
    Status AbstractTensorHandle::TensorHandleStatus() const {
      // Tensor handles in current runtime don't carry error info and this method
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Thu Feb 15 09:49:45 GMT 2024
    - 1.4K bytes
    - Viewed (0)
  9. tensorflow/c/eager/immediate_execution_tensor_handle.h

      virtual bool PreferCustomSummarizer() const { return false; }
    
      // Returns a string which summarizes the value of this TensorHandle, for
      // debugging. Does not include a shape or dtype.
      //
      // Included in the default implementation of DebugString.
      virtual Status SummarizeValue(std::string& summary) const;
    
      // For LLVM style RTTI.
      static bool classof(const AbstractTensorHandle* ptr) {
    C
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Fri Mar 10 21:56:24 GMT 2023
    - 4.3K bytes
    - Viewed (0)
  10. tensorflow/c/eager/unified_api_testutil.h

    namespace tensorflow {
    
    // Builds and returns a `TracingContext` using the default tracing impl.
    AbstractContext* BuildFunction(const char* fn_name);
    
    // Creates parameters (placeholders) in the tracing `ctx` using the shape and
    // dtype of `inputs`.
    Status CreateParamsForInputs(AbstractContext* ctx,
                                 absl::Span<AbstractTensorHandle* const> inputs,
                                 std::vector<AbstractTensorHandle*>* params);
    
    C
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Tue Feb 27 13:57:45 GMT 2024
    - 4K bytes
    - Viewed (0)
Back to top