Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/python/eager/
Dpywrap_tensor.cc147 bool IsCompatible(int desired_dtype, TF_DataType returned_dtype) { in IsCompatible() argument
149 static_cast<tensorflow::DataType>(desired_dtype); in IsCompatible()
212 int desired_dtype = -1; in ConvertToEagerTensor() local
214 if (!PyIntToDataType(dtype, &desired_dtype)) { in ConvertToEagerTensor()
233 if (desired_dtype >= 0) { in ConvertToEagerTensor()
235 static_cast<TF_DataType>(desired_dtype), &desired_np_dtype) in ConvertToEagerTensor()
239 "Invalid dtype argument value ", desired_dtype) in ConvertToEagerTensor()
405 int desired_dtype = -1; in EagerTensor_init() local
407 if (!PyIntToDataType(dtype, &desired_dtype)) { in EagerTensor_init()
422 if (desired_dtype >= 0 && desired_dtype != handle_dtype) { in EagerTensor_init()
[all …]
Dpywrap_tensor.h29 bool IsCompatible(int desired_dtype, TF_DataType returned_dtype);
Dpywrap_tfe_src.cc2182 const TF_DataType& desired_dtype, argument
2188 if (desired_dtype >= 0 && desired_dtype != input_dtype) {
2191 static_cast<TF_DataType>(desired_dtype), status));
2195 output_dtype = desired_dtype;
2308 int desired_dtype = -1; local
2310 if (!ParseTypeValue("", dtype_hint.get(), status, &desired_dtype)) {
2323 if (desired_dtype >= 0 && desired_dtype != output_dtype) {
2324 if (tensorflow::IsCompatible(desired_dtype, output_dtype)) {
2325 if (!CastTensor(op_exec_info, static_cast<TF_DataType>(desired_dtype),
2338 static_cast<tensorflow::DataType>(desired_dtype)))