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.cc39 TF_RETURN_IF_ERROR(ArgNumType(attrs, in, &is_type_list, &dtypes)); in ConstructFunctionGraph()
85 TF_RETURN_IF_ERROR(ArgNumType(attrs, out, &is_type_list, &dtypes)); in ConstructFunctionGraph()
/external/tensorflow/tensorflow/core/framework/
Dfunction.cc54 Status ArgNumType(AttrSlice attrs, const OpDef::ArgDef& arg_def, in ArgNumType() function
160 ArgNumType(attr_values, arg_def, &is_type_list, &dtypes)); in BuildInputArgIndex()
201 ArgNumType(attrs, node_sig->output_arg(i), &is_type_list, &dtypes)); in BuildNodeOutputIndex()
233 ArgNumType(attrs, fnode_sig->input_arg(i), &is_type_list, &dtypes)); in InstantiateNode()
321 TF_RETURN_IF_ERROR(ArgNumType(attrs, ret_def, &is_type_list, &dtypes)); in AddReturnNode()
Dfunction.h754 Status ArgNumType(AttrSlice attrs, const OpDef::ArgDef& arg_def,
/external/tensorflow/tensorflow/core/kernels/
Dpartitioned_function_ops.cc141 TF_RETURN_IF_ERROR(ArgNumType(attrs, ret_def, &is_type_list, &dtypes)); in FillOutputDevices()