Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 6 of 6 for tozero (0.36 sec)

  1. tensorflow/c/c_api.cc

        if (!status->status.ok()) return false;
      }
      return true;
    }
    
    // Create an empty tensor of type 'dtype'. 'shape' can be arbitrary, but has to
    // result in a zero-sized tensor.
    static TF_Tensor* EmptyTensor(TF_DataType dtype,
                                  const tensorflow::TensorShape& shape) {
      static char empty;
      int64_t nelems = 1;
      std::vector<int64_t> dims;
    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/c_api_cluster_test.cc

      TF_DeleteTensor(t);
    
      for (int i = 0; i < expected_values.size(); i++) {
        EXPECT_EQ(expected_values[i], actual_values[i])
            << "Mismatch in expected values at (zero-based) index " << i;
      }
    }
    
    void CheckRemoteMatMulExecutesOK(TFE_Context* ctx,
                                     const char* remote_device_name,
                                     const char* local_device_name) {
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Fri Apr 14 10:03:59 GMT 2023
    - 19.3K bytes
    - Viewed (0)
  3. tensorflow/c/eager/parallel_device/parallel_device.cc

      return result;
    }
    
    // Used as an argument to TFE_NewCustomDeviceTensorHandle, indicating how
    // ParallelTensors wrapped in TFE_TensorHandles should be cleaned up once their
    // reference counts drop to zero.
    void ParallelTensorDeallocator(void* data) {
      delete reinterpret_cast<ParallelTensor*>(data);
    }
    
    // Used as an argument to TFE_NewCustomDeviceTensorHandle, for computing the
    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)
  4. tensorflow/c/experimental/filesystem/plugins/gcs/ram_file_block_cache_test.cc

        EXPECT_EQ(got_n, want_n);
        calls++;
        memset(buffer, 'x', got_n);
        TF_SetStatus(status, TF_OK, "");
        return got_n;
      };
      // If block_size, max_bytes, or both are zero, or want_n is larger than
      // max_bytes the cache is a pass-through.
      tf_gcs_filesystem::RamFileBlockCache cache1(1, 0, 0, fetcher);
      tf_gcs_filesystem::RamFileBlockCache cache2(0, 1, 0, fetcher);
    C++
    - Registered: Tue Apr 23 12:39:09 GMT 2024
    - Last Modified: Fri Oct 15 03:16:57 GMT 2021
    - 23.2K bytes
    - Viewed (0)
  5. tensorflow/c/c_test_util.cc

                      const char* name, TF_Operation** op) {
      TF_Operation* zero = ScalarConst(
          0, graph, s, ::tensorflow::strings::StrCat(name, "_const0").c_str());
      TF_OperationDescription* desc = TF_NewOperation(graph, "Split", name);
      TF_AddInput(desc, {zero, 0});
      TF_AddInput(desc, {input, 0});
      TF_SetAttrInt(desc, "num_split", 3);
      TF_SetAttrType(desc, "T", TF_INT32);
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Fri Oct 15 03:16:52 GMT 2021
    - 17.8K bytes
    - Viewed (2)
  6. tensorflow/c/eager/dlpack.cc

      void* data = dl_tensor->data;
    
      if (dl_tensor->byte_offset != 0) {
        status->status = tensorflow::errors::InvalidArgument(
            "Unsupported byte_offset (", dl_tensor->byte_offset,
            ") from DLPack, must be zero");
        return nullptr;
      }
    
      size_t total_bytes = dl_tensor->dtype.bits / 8;
      for (int i = 0; i < num_dims; i++) {
        total_bytes *= dims[i];
      }
    
      if (dl_tensor->strides != nullptr &&
    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)
Back to top