Home
last modified time | relevance | path

Searched refs:ints_on_device (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/core/framework/
Dfunction.cc174 bool ints_on_device, int64 resource_arg_unique_id) { in BuildInputArgIndex() argument
193 if (ints_on_device && dtypes[i] == DataType::DT_INT32) { in BuildInputArgIndex()
340 bool ints_on_device, int* ret_index) { in AddReturnNode() argument
366 if (ints_on_device && dtypes[i] == DataType::DT_INT32) { in AddReturnNode()
726 bool ints_on_device = in InstantiateFunction() local
743 ints_on_device, resource_arg_unique_id); in InstantiateFunction()
799 s = helper.AddReturnNode(ret_def, attr_values, fdef.ret(), ints_on_device, in InstantiateFunction()
/external/tensorflow/tensorflow/compiler/tf2tensorrt/kernels/
Dtrt_engine_op.cc293 bool ints_on_device = in ConstructFunctionHandle() local
298 if (ints_on_device) { in ConstructFunctionHandle()