Home
last modified time | relevance | path

Searched refs:GetScalarFromTensor (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2xla/ops/
Dxla_ops.cc529 TF_RETURN_IF_ERROR(c->GetScalarFromTensor(padding_low_tensor, i, &low)); in __anonb720852f0302()
531 c->GetScalarFromTensor(padding_high_tensor, i, &high)); in __anonb720852f0302()
533 c->GetScalarFromTensor(padding_interior_tensor, i, &interior)); in __anonb720852f0302()
/external/tensorflow/tensorflow/core/framework/
Dshape_inference.cc925 Status InferenceContext::GetScalarFromTensor(const Tensor* t, int64_t* val) { in GetScalarFromTensor() function in tensorflow::shape_inference::InferenceContext
943 Status InferenceContext::GetScalarFromTensor(const Tensor* t, int64_t idx, in GetScalarFromTensor() function in tensorflow::shape_inference::InferenceContext
980 TF_RETURN_IF_ERROR(GetScalarFromTensor(t, &val)); in MakeDimForScalarInput()
997 TF_RETURN_IF_ERROR(GetScalarFromTensor(t, &val)); in MakeDimForScalarInputWithNegativeIndexing()
Dshape_inference.h581 Status GetScalarFromTensor(const Tensor* t, int64_t* val);
585 Status GetScalarFromTensor(const Tensor* t, int64_t idx, int64_t* val);
Dcommon_shape_fns.cc1459 TF_RETURN_IF_ERROR(c->GetScalarFromTensor(num_rows_tensor, &num_rows)); in MatrixDiagV2Shape()
1462 TF_RETURN_IF_ERROR(c->GetScalarFromTensor(num_cols_tensor, &num_cols)); in MatrixDiagV2Shape()
2617 TF_RETURN_IF_ERROR(c->GetScalarFromTensor(scatter_dimension, &scatter_dim)); in ReduceScatterShape()
/external/tensorflow/tensorflow/core/ops/
Dmath_ops.cc1687 TF_RETURN_IF_ERROR(c->GetScalarFromTensor(nbins_input, &nbins)); in __anon328d24420d02()