Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/
H A Dlayout_util.cc67 TF_ASSIGN_OR_RETURN(DataType dtype, EncodePrimitiveTypeAsDataType( in RewriteLayoutWithShardedShape()
102 TF_ASSIGN_OR_RETURN(DataType dtype, EncodePrimitiveTypeAsDataType( in ReshapeWithCorrectRepresentationAndSharding()
H A Dtype_util.h37 StatusOr<DataType> EncodePrimitiveTypeAsDataType(xla::PrimitiveType type);
H A Dtype_util.cc83 StatusOr<DataType> EncodePrimitiveTypeAsDataType(xla::PrimitiveType type) { in EncodePrimitiveTypeAsDataType() function
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/
H A Dcompile_mlir_util.cc195 TF_ASSIGN_OR_RETURN(DataType dtype, EncodePrimitiveTypeAsDataType( in GetOutputInfo()
531 TF_ASSIGN_OR_RETURN(DataType dtype, EncodePrimitiveTypeAsDataType( in ConvertMLIRToXlaComputation()
541 TF_ASSIGN_OR_RETURN(DataType dtype, EncodePrimitiveTypeAsDataType( in ConvertMLIRToXlaComputation()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
H A Dimage_resize_ops.cc516 EncodePrimitiveTypeAsDataType(input_type).ValueOrDie(); in GeneralCompile()