Searched refs:ArgNumType (Results 1 – 4 of 4) sorted by relevance
/external/tensorflow/tensorflow/core/distributed_runtime/ |
D | cluster_function_library_runtime.cc | 66 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/ |
D | partitioned_function_ops.cc | 146 TF_RETURN_IF_ERROR(ArgNumType(attrs, ret_def, &is_type_list, &dtypes)); in FillOutputDevices()
|
/external/tensorflow/tensorflow/core/framework/ |
D | function.cc | 69 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()
|
D | function.h | 879 Status ArgNumType(AttrSlice attrs, const OpDef::ArgDef& arg_def,
|