Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 3 of 3 for choice (0.26 sec)

  1. tensorflow/c/eager/dlpack.cc

      // There are two ways to represent compact row-major data
      // 1) nullptr indicates tensor is compact and row-majored.
      // 2) fill in the strides array as the real case for compact row-major data.
      // Here we choose option 2, since some frameworks didn't handle the strides
      // argument properly.
      dlm_tensor->dl_tensor.strides = stride_arr->data();
    
      dlm_tensor->dl_tensor.byte_offset =
    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)
  2. tensorflow/c/eager/c_api.cc

        return status.status;
      }
    
      absl::StatusOr<bool> ShallPinToThisDevice(
          const ImmediateExecutionOperation* op) override {
        TF_Status status;
        // Let this custom device choose the device to pin this op on if it
        // implements the pinning function.
        if (device_.shall_pin_to_this_device != nullptr) {
          return device_.shall_pin_to_this_device(tensorflow::wrap(op), &status);
        }
    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)
  3. tensorflow/c/eager/c_api_distributed_test.cc

      TFE_ContextSetServerDef(ctx, 0, serialized.data(), serialized.size(), status);
      EXPECT_EQ(TF_OK, TF_GetCode(status)) << TF_Message(status);
    
      // Use large matrices so that RPCs don't return before we get a chance
      // to call TFE_DeleteContext.
      TFE_TensorHandle* h0_task0 = TestMatrixTensorHandle100x100(ctx);
      TFE_TensorHandle* h1_task0 = TestMatrixTensorHandle100x100(ctx);
      const char remote_device_name[] =
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Thu Feb 15 09:49:45 GMT 2024
    - 23.5K bytes
    - Viewed (0)
Back to top