Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 10 of 24 for dlim (0.13 sec)

  1. tensorflow/c/c_api_internal.h

          :
    #if !defined(IS_MOBILE_PLATFORM) && !defined(IS_SLIM_BUILD)
            api_def_map(op_list),
    #endif  // !defined(IS_MOBILE_PLATFORM) && !defined(IS_SLIM_BUILD)
            update_docs_called(false) {
      }
    
    #if !defined(IS_MOBILE_PLATFORM) && !defined(IS_SLIM_BUILD)
      tensorflow::ApiDefMap api_def_map TF_GUARDED_BY(lock);
    #endif  // !defined(IS_MOBILE_PLATFORM) && !defined(IS_SLIM_BUILD)
      bool update_docs_called TF_GUARDED_BY(lock);
    C
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Sat May 13 00:49:12 GMT 2023
    - 7.6K bytes
    - Viewed (0)
  2. tensorflow/c/eager/immediate_execution_tensor_handle.h

      // ImmediateExecutionTensorHandles, but comes up for example when computing
      // the shape of a parallel tensor with component shapes differing across
      // devices.
      virtual Status Dim(int dim_index, int64_t* dim) const = 0;
    
      // Returns the device which created the handle.
      virtual const char* DeviceName(Status* status) const = 0;
      // Returns the device where the tensor was placed.
    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)
  3. tensorflow/c/c_api_test.cc

    #undef EXPECT_TF_META
    
    TEST(CAPI, TestTensorAligned) {
      int64_t dim = 7;
      size_t tensor_size_bytes = dim * TF_DataTypeSize(TF_FLOAT);
      TF_Tensor* a = TF_AllocateTensor(
          /*dtype=*/TF_FLOAT, /*dims=*/&dim, /*num_dims=*/1,
          /*len=*/tensor_size_bytes);
      float* data = reinterpret_cast<float*>(TF_TensorData(a));
      for (int i = 0; i < dim; ++i) {
        data[i] = 0;
      }
      if (TF_TensorDefaultAlignment() > 0) {
    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)
  4. tensorflow/c/eager/c_api_test.cc

      ASSERT_EQ(num_dims, -1);
      ASSERT_EQ("Invalid handle", string(TF_Message(status.get())));
    
      TF_SetStatus(status.get(), TF_OK, "");
    
      int dim = TFE_TensorHandleDim(h, 0, status.get());
      ASSERT_EQ(TF_INVALID_ARGUMENT, TF_GetCode(status.get()));
      ASSERT_EQ(dim, -1);
      ASSERT_EQ("Invalid handle", string(TF_Message(status.get())));
    }
    
    TEST(CAPI, TensorHandleDevices) {
    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)
  5. tensorflow/c/c_api.cc

    #if !defined(IS_MOBILE_PLATFORM) && !defined(IS_SLIM_BUILD)
      delete server;
    #endif  // !defined(IS_MOBILE_PLATFORM) && !defined(IS_SLIM_BUILD)
    }
    
    void TF_RegisterLogListener(void (*listener)(const char*)) {
    #if !defined(IS_MOBILE_PLATFORM) && !defined(IS_SLIM_BUILD)
      tensorflow::logging::RegisterListener(listener);
    #endif  // !defined(IS_MOBILE_PLATFORM) && !defined(IS_SLIM_BUILD)
    }
    
    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/dlpack.cc

      auto* tf_dlm_tensor_ctx = new TfDlManagedTensorCtx(tensor_ref);
      tf_dlm_tensor_ctx->reference = tensor_ref;
    
      DLManagedTensor* dlm_tensor = &tf_dlm_tensor_ctx->tensor;
      dlm_tensor->manager_ctx = tf_dlm_tensor_ctx;
      dlm_tensor->deleter = &DLManagedTensorDeleter;
      dlm_tensor->dl_tensor.device = tf_dlm_context;
      int ndim = tensor->dims();
      dlm_tensor->dl_tensor.ndim = ndim;
      dlm_tensor->dl_tensor.data = tf_dlm_data;
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Thu Feb 15 09:49:45 GMT 2024
    - 12.8K bytes
    - Viewed (0)
  7. tensorflow/c/eager/dlpack_test.cc

        data[j] = j;
      }
      DLManagedTensor dlm_in = {};
      DLTensor* dltensor_in = &dlm_in.dl_tensor;
      dltensor_in->data = data.data();
      dltensor_in->device = {kDLCPU, 0};
      dltensor_in->ndim = static_cast<int32_t>(shape.size());
      dltensor_in->dtype = {kDLFloat, 32, 1};
      dltensor_in->shape = shape.data();
      dltensor_in->strides = strides.data();
      TFE_TensorHandle* handle = TFE_HandleFromDLPack(&dlm_in, status, ctx);
      ASSERT_NE(handle, nullptr)
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Fri Jun 30 03:04:46 GMT 2023
    - 4.4K bytes
    - Viewed (0)
  8. tensorflow/c/experimental/gradients/nn_grad.cc

      }
      auto imm_ctx = dyn_cast<ImmediateExecutionContext>(ctx);
      AbstractTensorPtr minus_1(imm_ctx->CreateInt32Scalar(-1));
      ImmediateTensorHandlePtr dim(imm_ctx->CreateLocalHandle(minus_1.get()));
      AbstractTensorHandle* expand_dims_outputs;
      TF_RETURN_IF_ERROR(
          ops::ExpandDims(ctx, vec, dim.get(), &expand_dims_outputs, "ExpandDims"));
      TF_RETURN_IF_ERROR(
          ops::Mul(ctx, expand_dims_outputs, mat, &outputs[0], "Mul"));
    C++
    - Registered: Tue Mar 26 12:39:09 GMT 2024
    - Last Modified: Wed Feb 28 13:53:47 GMT 2024
    - 5.7K bytes
    - Viewed (0)
  9. tensorflow/c/eager/c_api_unified_experimental.cc

        return nullptr;
      }
      tensorflow::PartialTensorShape partial_shape;
      if (shape.num_dims != -1) {
        DCHECK(shape.dim_sizes != nullptr);
        Status status = tensorflow::PartialTensorShape::MakePartialShape(
            reinterpret_cast<int64_t*>(shape.dim_sizes), shape.num_dims,
            &partial_shape);
        if (!status.ok()) {
          tsl::Set_TF_Status_from_Status(s, status);
          return nullptr;
        }
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Thu Feb 15 09:49:45 GMT 2024
    - 9K bytes
    - Viewed (0)
  10. tensorflow/c/eager/c_api.h

    TF_CAPI_EXPORT extern int TFE_TensorDebugInfoOnDeviceNumDims(
        TFE_TensorDebugInfo* debug_info);
    
    // Returns the number of elements in dimension `dim_index`.
    // Tensor representation on device can be transposed from its representation
    // on host. The data contained in dimension `dim_index` on device
    // can correspond to the data contained in another dimension in on-host
    // representation. The dimensions are indexed using the standard TensorFlow
    C
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Thu Apr 27 21:07:00 GMT 2023
    - 22.8K bytes
    - Viewed (1)
Back to top