Home	Sort by relevance Sort by last modified time

Full Search Definition Symbol File Path History   | | Help

    Searched refs:DataTypeToPrimitiveType (Results 1 - 16 of 16) sorted by null

/external/tensorflow/tensorflow/compiler/tf2xla/
type_util.h	26 Status DataTypeToPrimitiveType(DataType data_type, xla::PrimitiveType* type);
xla_helpers.cc	50 TF_RETURN_IF_ERROR(DataTypeToPrimitiveType(output_type, &xla_output_type)); 96 TF_CHECK_OK(DataTypeToPrimitiveType(data_type, &type)); 103 TF_CHECK_OK(DataTypeToPrimitiveType(data_type, &type)); 110 TF_CHECK_OK(DataTypeToPrimitiveType(data_type, &type)); 117 TF_CHECK_OK(DataTypeToPrimitiveType(data_type, &type)); 139 TF_CHECK_OK(DataTypeToPrimitiveType(data_type, &type)); 147 TF_CHECK_OK(DataTypeToPrimitiveType(data_type, &type));
literal_util.cc	50 DataTypeToPrimitiveType(host_tensor->dtype(), &primitive_type));
shape_util.cc	54 TF_RETURN_IF_ERROR(DataTypeToPrimitiveType(dtype, &type));
type_util.cc	23 Status DataTypeToPrimitiveType(DataType data_type, xla::PrimitiveType* type) { 75 "Unsupported type in DataTypeToPrimitiveType ",
xla_context.cc	132 TF_CHECK_OK(DataTypeToPrimitiveType(type, &xla_type)); 146 TF_CHECK_OK(DataTypeToPrimitiveType(type, &xla_type)); 160 TF_CHECK_OK(DataTypeToPrimitiveType(type, &xla_type)); 174 TF_CHECK_OK(DataTypeToPrimitiveType(type, &xla_type));
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
cast_op.cc	31 OP_REQUIRES_OK(ctx, DataTypeToPrimitiveType(src_dtype_, &src_type_)); 32 OP_REQUIRES_OK(ctx, DataTypeToPrimitiveType(dst_dtype_, &dst_type_));
batch_norm_op.cc	46 DataTypeToPrimitiveType(ctx->input_type(0), &input_type)); 49 DataTypeToPrimitiveType(ctx->input_type(1), &scale_type)); 136 OP_REQUIRES_OK(ctx, DataTypeToPrimitiveType(input_dtype, &input_type)); 138 OP_REQUIRES_OK(ctx, DataTypeToPrimitiveType(scale_dtype, &scale_type));
reduction_ops.cc	66 TF_CHECK_OK(DataTypeToPrimitiveType(input_type(0), &type)); 86 TF_CHECK_OK(DataTypeToPrimitiveType(input_type(0), &type));
categorical_op.cc	65 DataTypeToPrimitiveType(input_type(0), &uniform_xla_type));
reduction_ops_common.cc	109 TF_CHECK_OK(DataTypeToPrimitiveType(input_type(0), &type));
extract_image_patches_op.cc	87 OP_REQUIRES_OK(ctx, DataTypeToPrimitiveType(ctx->input_type(0), &type));
gather_op.cc	157 TF_RETURN_IF_ERROR(DataTypeToPrimitiveType(index_type, &ptype));
diag_op.cc	74 DataTypeToPrimitiveType(ctx->input_type(0), &element_type));
image_resize_ops.cc	368 OP_REQUIRES_OK(ctx, DataTypeToPrimitiveType(output_dtype, &output_type_));
pooling_ops.cc	342 OP_REQUIRES_OK(ctx, DataTypeToPrimitiveType(input_type(2), &element_type));

Completed in 307 milliseconds