Home
last modified time | relevance | path

Searched refs:ArgNumType (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/core/distributed_runtime/
Dcluster_function_library_runtime.cc66 TF_RETURN_IF_ERROR(ArgNumType(attrs, in, &is_type_list, &dtypes)); in ConstructFunctionGraph()
118 TF_RETURN_IF_ERROR(ArgNumType(attrs, out, &is_type_list, &dtypes)); in ConstructFunctionGraph()
/external/tensorflow/tensorflow/core/kernels/
Dpartitioned_function_ops.cc146 TF_RETURN_IF_ERROR(ArgNumType(attrs, ret_def, &is_type_list, &dtypes)); in FillOutputDevices()
/external/tensorflow/tensorflow/core/framework/
Dfunction.cc69 Status ArgNumType(AttrSlice attrs, const OpDef::ArgDef& arg_def, in ArgNumType() function
178 ArgNumType(attr_values, arg_def, &is_type_list, &dtypes)); in BuildInputArgIndex()
228 ArgNumType(attrs, node_sig->output_arg(i), &is_type_list, &dtypes)); in BuildNodeOutputIndex()
260 ArgNumType(attrs, fnode_sig->input_arg(i), &is_type_list, &dtypes)); in InstantiateNode()
347 TF_RETURN_IF_ERROR(ArgNumType(attrs, ret_def, &is_type_list, &dtypes)); in AddReturnNode()
Dfunction.h879 Status ArgNumType(AttrSlice attrs, const OpDef::ArgDef& arg_def,