Searched refs:GetScalarFromTensor (Results 1 – 4 of 4) sorted by relevance
888 Status InferenceContext::GetScalarFromTensor(const Tensor* t, int64* val) { in GetScalarFromTensor() function in tensorflow::shape_inference::InferenceContext914 TF_RETURN_IF_ERROR(GetScalarFromTensor(t, &val)); in MakeDimForScalarInput()931 TF_RETURN_IF_ERROR(GetScalarFromTensor(t, &val)); in MakeDimForScalarInputWithNegativeIndexing()
474 Status GetScalarFromTensor(const Tensor* t, int64* val);
1315 TF_RETURN_IF_ERROR(c->GetScalarFromTensor(num_rows_tensor, &num_rows)); in MatrixDiagV2Shape()1318 TF_RETURN_IF_ERROR(c->GetScalarFromTensor(num_cols_tensor, &num_cols)); in MatrixDiagV2Shape()
1596 TF_RETURN_IF_ERROR(c->GetScalarFromTensor(nbins_input, &nbins)); in __anonb5dc68880d02()