Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 10 of 16 for dlim (0.15 sec)

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

      auto num_dims_analytical = TF_NumDims(analytical_tensor);
      ASSERT_EQ(dims.size(), num_dims_analytical);
      for (int j = 0; j < num_dims_analytical; j++) {
        auto dim_analytical = TF_Dim(analytical_tensor, j);
        ASSERT_EQ(dims[j], dim_analytical);
        num_elem_analytical *= dim_analytical;
      }
    
      float* danalytical = new float[num_elem_analytical]{0};
      memcpy(&danalytical[0], TF_TensorData(analytical_tensor),
    C++
    - Registered: Tue Mar 26 12:39:09 GMT 2024
    - Last Modified: Wed Feb 28 13:53:47 GMT 2024
    - 5K bytes
    - Viewed (0)
  6. tensorflow/c/c_api_experimental.cc

    // for the first time.
    TF_Library* TF_LoadPluggableDeviceLibrary(const char* library_filename,
                                              TF_Status* status) {
    #if defined(IS_MOBILE_PLATFORM) || defined(IS_SLIM_BUILD)
      status->status = tensorflow::errors::Unimplemented(
          "PluggableDevice plugin functionality is not supported on mobile");
      return nullptr;
    #else
      TF_Library* lib_handle = new TF_Library;
    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)
  7. 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)
  8. tensorflow/c/eager/gradients.cc

      }
      TensorShapeProto proto;
      if (num_dims < 0) {
        proto.set_unknown_rank(true);
      } else {
        for (int d = 0; d < num_dims; ++d) {
          proto.add_dim()->set_size(dims[d]);
        }
      }
    
      forward_op_->attrs.Set(attr_name, proto);
      return op_->SetAttrShape(attr_name, dims, num_dims);
    }
    Status SetAttrFunction(AbstractOperation* op_, const char* attr_name,
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Thu Feb 15 09:49:45 GMT 2024
    - 19.3K bytes
    - Viewed (0)
  9. tensorflow/c/eager/parallel_device/parallel_device_lib.cc

              int64_t axis_length = combined_shape.dim_size(axis_index);
              if (axis_length != component_shape.dim_size(axis_index)) {
                axis_length = -1;
              }
              TF_RETURN_IF_ERROR(
                  combined_shape.SetDimWithStatus(axis_index, axis_length));
            }
          }
        }
        auto dim_sizes = combined_shape.dim_sizes();
        shape_ = std::vector<int64_t>(dim_sizes.begin(), dim_sizes.end());
      }
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Fri Feb 09 07:47:20 GMT 2024
    - 25.4K bytes
    - Viewed (1)
  10. tensorflow/c/eager/parallel_device/parallel_device.cc

    // dimension of a parallel tensor.
    int64_t ParallelTensorDim(void* data, int dim_index, TF_Status* status) {
      const std::vector<int64_t>* shape;
      Status s = reinterpret_cast<ParallelTensor*>(data)->Shape(&shape);
      if (!s.ok()) {
        tsl::Set_TF_Status_from_Status(status, s);
        return -1;
      }
      return (*shape)[dim_index];
    }
    
    TF_Buffer* ParallelTensorSummarize(void* data, TF_Status* status) {
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Wed Mar 29 22:05:31 GMT 2023
    - 18.3K bytes
    - Viewed (0)
Back to top