Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 10 of 19 for dtype (0.14 sec)

  1. 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)
  2. 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)
  3. 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)
  4. 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)
  5. tensorflow/c/c_api.cc

        for (const auto& p : *shapes_and_types) {
          auto* out_shape_and_type = handle_data.add_shape_and_type();
          ic->ShapeHandleToProto(p.shape, out_shape_and_type->mutable_shape());
          out_shape_and_type->set_dtype(p.dtype);
          *out_shape_and_type->mutable_type() = p.type;
        }
      }
      string str_data;
      handle_data.SerializeToString(&str_data);
    
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Mon Apr 15 03:35:10 GMT 2024
    - 102.3K bytes
    - Viewed (0)
  6. tensorflow/c/eager/c_api_unified_experimental_graph.cc

            operation->SetOpName(absl::StrCat("_input_", inputs_.size()).c_str()));
        TF_RETURN_IF_ERROR(operation->SetAttrType("dtype", dtype));
        if (!shape.unknown_rank()) {
          TF_RETURN_IF_ERROR(operation->SetAttrShape(
              "shape", reinterpret_cast<int64_t*>(shape.dim_sizes().data()),
              shape.dims()));
        }
        int num_outputs = 1;
        std::vector<AbstractTensorHandle*> outputs(num_outputs);
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Tue Mar 12 20:00:09 GMT 2024
    - 15.4K bytes
    - Viewed (1)
  7. tensorflow/c/eager/c_api.cc

          }
          // Type
          if (const int type_size = default_value.list().type_size()) {
            absl::InlinedVector<unsigned int, 4> type_vector;
            type_vector.reserve(type_size);
            for (int i = 0; i < type_size; ++i) {
              type_vector.push_back(default_value.list().type(i));
            }
            TFE_OpSetAttrTypeList(
                op, attr_name,
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Tue Mar 12 20:00:09 GMT 2024
    - 43.9K bytes
    - Viewed (2)
  8. tensorflow/c/eager/c_api_experimental.cc

      }
    
      tensorflow::AbstractTensorInterface* t =
          tensorflow::unwrap(ctx)->CreateTensor(
              static_cast<tensorflow::DataType>(dtype), dimvec);
    
      if (t == nullptr) {
        status->status =
            tensorflow::errors::InvalidArgument("Unsupported dtype: ", dtype);
        return nullptr;
      }
    
      return new TF_Tensor{t};
    }
    
    TFE_TensorHandle* TFE_NewTensorHandleFromTensor(TFE_Context* ctx, TF_Tensor* t,
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Thu Apr 11 23:52:39 GMT 2024
    - 35.9K bytes
    - Viewed (3)
  9. tensorflow/c/c_api_test.cc

      TF_OperationDescription* init(string type) {
        // Construct op_name to match the name used by REGISTER_OP in the
        // ATTR_TEST_REGISTER calls above.
        string op_name = "CApiAttributesTestOp";
        if (type.find("list(") == 0) {
          op_name += "List";
          type = type.replace(0, 5, "");
          type = type.replace(type.size() - 1, 1, "");
        }
        op_name += type;
        return TF_NewOperation(
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Mon Apr 15 03:35:10 GMT 2024
    - 96.9K bytes
    - Viewed (3)
  10. RELEASE.md

            the layer's dtype. If you had a model that used `float64`, it will
            probably silently use `float32` in TensorFlow2, and a warning will be
            issued that starts with Layer "layer-name" is casting an input tensor
            from dtype float64 to the layer's dtype of float32. To fix, either set
            the default dtype to float64 with
    Plain Text
    - Registered: Tue May 07 12:40:20 GMT 2024
    - Last Modified: Mon Apr 29 19:17:57 GMT 2024
    - 727.7K bytes
    - Viewed (8)
Back to top