Searched refs:TF_TensorElementCount (Results 1 – 10 of 10) sorted by relevance
93 for (int i = 0; i < TF_TensorElementCount(params.tags); ++i) { in ScalarSummaryOp_Compute()126 if (TF_TensorElementCount(tags) == 1) { in SingleTag()
64 for (int i = 0; i < TF_TensorElementCount(safe_input_ptr.get()); ++i) { in MergeSummaryOp_Compute()
90 for (int64_t i = 0; i < TF_TensorElementCount(safe_values_ptr.get()); ++i) { in HistogramSummaryOp_Compute()
51 auto num_elem_numerical = TF_TensorElementCount(numerical_tensor); in CompareNumericalAndAutodiffGradients()56 auto num_elem_analytical = TF_TensorElementCount(analytical_tensor); in CompareNumericalAndAutodiffGradients()
134 TF_CAPI_EXPORT extern int64_t TF_TensorElementCount(const TF_Tensor* tensor);
166 int64_t TF_TensorElementCount(const TF_Tensor* t) { in TF_TensorElementCount() function
173 EXPECT_EQ(6, TF_TensorElementCount(t)); in TEST()1521 EXPECT_EQ(6, TF_TensorElementCount(a)); in TEST()1522 EXPECT_EQ(1, TF_TensorElementCount(b)); in TEST()1532 EXPECT_EQ(6, TF_TensorElementCount(a)); in TEST()1533 EXPECT_EQ(6, TF_TensorElementCount(b)); in TEST()
140 return TF_TensorElementCount(tensor_.get()); in num_elements()
51 auto num_elem_numerical = TF_TensorElementCount(numerical_tensor); in CompareNumericalAndManualGradients()
109 int num_elems = TF_TensorElementCount(theta_tensor); in CalcNumericalGrad()