Searched refs:GetScalarFromTensor (Results 1 – 5 of 5) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/ops/ |
| H A D | xla_ops.cc | 529 TF_RETURN_IF_ERROR(c->GetScalarFromTensor(padding_low_tensor, i, &low)); in __anon6b9e6b800302()
531 c->GetScalarFromTensor(padding_high_tensor, i, &high)); in __anon6b9e6b800302()
533 c->GetScalarFromTensor(padding_interior_tensor, i, &interior)); in __anon6b9e6b800302()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/framework/ |
| H A D | shape_inference.cc | 925 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()
|
| H A D | shape_inference.h | 581 Status GetScalarFromTensor(const Tensor* t, int64_t* val);
585 Status GetScalarFromTensor(const Tensor* t, int64_t idx, int64_t* val);
|
| H A D | common_shape_fns.cc | 1459 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()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/ops/ |
| H A D | math_ops.cc | 1687 TF_RETURN_IF_ERROR(c->GetScalarFromTensor(nbins_input, &nbins)); in __anon247ac1530d02()
|