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()
|