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