Searched refs:GetScalarFromTensor (Results 1 – 3 of 3) sorted by relevance
948 Status InferenceContext::GetScalarFromTensor(const Tensor* t, int64* val) { in GetScalarFromTensor() function in tensorflow::shape_inference::InferenceContext974 TF_RETURN_IF_ERROR(GetScalarFromTensor(t, &val)); in MakeDimForScalarInput()991 TF_RETURN_IF_ERROR(GetScalarFromTensor(t, &val)); in MakeDimForScalarInputWithNegativeIndexing()
509 Status GetScalarFromTensor(const Tensor* t, int64* val);
1503 TF_RETURN_IF_ERROR(c->GetScalarFromTensor(nbins_input, &nbins)); in __anon57f5a5dd0d02()