Home
last modified time | relevance | path

Searched refs:GetElementTypeFromAccess (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dcollection_ops_util.cc172 llvm::Optional<RankedTensorType> GetElementTypeFromAccess( in GetElementTypeFromAccess() function
179 auto type_from_body = GetElementTypeFromAccess( in GetElementTypeFromAccess()
186 auto type_from_then = GetElementTypeFromAccess( in GetElementTypeFromAccess()
190 auto type_from_else = GetElementTypeFromAccess( in GetElementTypeFromAccess()
196 auto type_from_callee = GetElementTypeFromAccess( in GetElementTypeFromAccess()
200 auto type_from_alias = GetElementTypeFromAccess( in GetElementTypeFromAccess()
Dcollection_ops_util.h81 llvm::Optional<RankedTensorType> GetElementTypeFromAccess(
Dstack_ops_decomposition.cc380 auto elem_type = cutil::GetElementTypeFromAccess( in HandleStackV2Op()
Dtensor_array_ops_decomposition.cc117 auto elem_type = cutil::GetElementTypeFromAccess( in GetTensorArrayElementShape()