Searched refs:OpDef_ArgDef (Results 1 – 9 of 9) sorted by relevance
/external/tensorflow/tensorflow/core/grappler/optimizers/data/ |
D | function_utils.cc | 105 OpDef_ArgDef* AddFunctionInput(const string& name, FunctionDef* fdef, in AddFunctionInput() 150 [&name](const OpDef_ArgDef& arg) { return arg.name() == name; }, in FindFunctionInputWithName() 156 [&name](const OpDef_ArgDef& arg) { return arg.name() == name; }, in FindFunctionOutputWithName()
|
D | function_utils.h | 67 OpDef_ArgDef* AddFunctionInput(const string& name, FunctionDef* fdef,
|
D | hoist_random_uniform.cc | 133 OpDef_ArgDef* InsertSeedArgument(FunctionDef* function, int num_placeholders) { in InsertSeedArgument()
|
D | fusion_utils.cc | 262 const protobuf::RepeatedPtrField<OpDef_ArgDef>& fake_args, in CreateFalsePredicate()
|
/external/tensorflow/tensorflow/java/src/gen/cc/ |
D | op_specs.cc | 53 Type TypeOf(const OpDef_ArgDef& arg_def, bool* iterable_out); 92 Type TypeResolver::TypeOf(const OpDef_ArgDef& arg_def, bool* iterable_out) { in TypeOf() 285 ArgumentSpec CreateInput(const OpDef_ArgDef& input_def, in CreateInput() 320 ArgumentSpec CreateOutput(const OpDef_ArgDef& output_def, in CreateOutput()
|
/external/tensorflow/tensorflow/go/genop/internal/ |
D | genop.go | 376 op *odpb.OpDef_ArgDef 559 func isListArg(argdef *odpb.OpDef_ArgDef) bool { argument
|
/external/tensorflow/tensorflow/js/ops/ |
D | ts_op_gen.cc | 28 static bool IsListAttr(const OpDef_ArgDef& arg) { in IsListAttr()
|
/external/tensorflow/tensorflow/compiler/tf2xla/ |
D | tf2xla_util.cc | 280 const OpDef_ArgDef& output_arg = body_func->signature().output_arg(i); in PropagateConstIntoWhileNode() 287 const OpDef_ArgDef& input_arg = body_func->signature().input_arg(i); in PropagateConstIntoWhileNode()
|
/external/tensorflow/tensorflow/c/ |
D | c_api_experimental.cc | 456 const tensorflow::OpDef_ArgDef& input_arg = op_def->input_arg()[input_index]; in TF_GetNumberAttrForOpListInput()
|