Searched refs:TFE_TensorHandleNumElements (Results 1 – 6 of 6) sorted by relevance
79 auto num_elements = TFE_TensorHandleNumElements(tensor, status); in ExtractSmallTensorValue()
1921 int64_t num_elements = TFE_TensorHandleNumElements(input, status); in Execute()
148 TF_CAPI_EXPORT extern int64_t TFE_TensorHandleNumElements(TFE_TensorHandle* h,
180 EXPECT_EQ(TFE_TensorHandleNumElements(packed_handle, status), 1); in TestFunctionWithPackedInput()
304 int64_t TFE_TensorHandleNumElements(TFE_TensorHandle* h, TF_Status* status) { in TFE_TensorHandleNumElements() function
643 int n = TFE_TensorHandleNumElements(handle, &self->status); in EagerTensor_num_elements()1006 int64_t result = TFE_TensorHandleNumElements(as_c_eager_tensor->handle, in PyEagerTensor_NumElements()