Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/tf2xla/
Dtype_util.h37 xla::StatusOr<DataType> EncodePrimitiveTypeAsDataType(xla::PrimitiveType type);
Dxla_helpers.cc168 TF_ASSIGN_OR_RETURN(DataType dtype, EncodePrimitiveTypeAsDataType( in RewriteLayoutWithShardedShape()
200 TF_ASSIGN_OR_RETURN(DataType dtype, EncodePrimitiveTypeAsDataType( in ReshapeWithCorrectRepresentationAndSharding()
Dtype_util.cc83 xla::StatusOr<DataType> EncodePrimitiveTypeAsDataType(xla::PrimitiveType type) { in EncodePrimitiveTypeAsDataType() function