HomeSort by relevance Sort by last modified time
    Searched refs:TF_Dim (Results 1 - 6 of 6) sorted by null

  /external/tensorflow/tensorflow/python/lib/core/
ndarray_tensor.cc 275 dims[i] = TF_Dim(tensor, i);
  /external/tensorflow/tensorflow/c/
c_api_test.cc 135 EXPECT_EQ(dims[0], TF_Dim(t, 0));
136 EXPECT_EQ(dims[1], TF_Dim(t, 1));
160 EXPECT_EQ(dims[0], TF_Dim(t, 0));
161 EXPECT_EQ(dims[1], TF_Dim(t, 1));
    [all...]
c_api.h 266 TF_CAPI_EXPORT extern int64_t TF_Dim(const TF_Tensor* tensor, int dim_index);
    [all...]
c_api.cc 243 int64_t TF_Dim(const TF_Tensor* t, int dim_index) {
    [all...]
  /external/tensorflow/tensorflow/java/src/main/native/
tensor_jni.cc 508 dims[i] = static_cast<jlong>(TF_Dim(t, i));
604 num_elements *= TF_Dim(t, i);
  /external/tensorflow/tensorflow/go/
tensor.go 151 shape[i] = int64(C.TF_Dim(c, C.int(i)))

Completed in 215 milliseconds