Searched refs:GetScalarFromTensor (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/ops/ |
| D | xla_ops.cc | 453 TF_RETURN_IF_ERROR(c->GetScalarFromTensor(padding_low_tensor, i, &low)); in __anonbbc1c2d10302()
455 c->GetScalarFromTensor(padding_high_tensor, i, &high)); in __anonbbc1c2d10302()
457 c->GetScalarFromTensor(padding_interior_tensor, i, &interior)); in __anonbbc1c2d10302()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | shape_inference.h | 495 Status GetScalarFromTensor(const Tensor* t, int64* val);
499 Status GetScalarFromTensor(const Tensor* t, int64 idx, int64* val);
|
| D | shape_inference.cc | 892 Status InferenceContext::GetScalarFromTensor(const Tensor* t, int64* val) { in GetScalarFromTensor() function in tensorflow::shape_inference::InferenceContext
910 Status InferenceContext::GetScalarFromTensor(const Tensor* t, int64 idx, in GetScalarFromTensor() function in tensorflow::shape_inference::InferenceContext
947 TF_RETURN_IF_ERROR(GetScalarFromTensor(t, &val)); in MakeDimForScalarInput()
964 TF_RETURN_IF_ERROR(GetScalarFromTensor(t, &val)); in MakeDimForScalarInputWithNegativeIndexing()
|
| D | common_shape_fns.cc | 1368 TF_RETURN_IF_ERROR(c->GetScalarFromTensor(num_rows_tensor, &num_rows)); in MatrixDiagV2Shape()
1371 TF_RETURN_IF_ERROR(c->GetScalarFromTensor(num_cols_tensor, &num_cols)); in MatrixDiagV2Shape()
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | math_ops.cc | 1629 TF_RETURN_IF_ERROR(c->GetScalarFromTensor(nbins_input, &nbins)); in __anon725c8f640d02()
|