Home
last modified time | relevance | path

Searched refs:OpDef_ArgDef (Results 1 – 9 of 9) sorted by relevance

/external/tensorflow/tensorflow/core/grappler/optimizers/data/
Dfunction_utils.cc105 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()
Dfunction_utils.h67 OpDef_ArgDef* AddFunctionInput(const string& name, FunctionDef* fdef,
Dhoist_random_uniform.cc133 OpDef_ArgDef* InsertSeedArgument(FunctionDef* function, int num_placeholders) { in InsertSeedArgument()
Dfusion_utils.cc262 const protobuf::RepeatedPtrField<OpDef_ArgDef>& fake_args, in CreateFalsePredicate()
/external/tensorflow/tensorflow/java/src/gen/cc/
Dop_specs.cc53 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/
Dgenop.go376 op *odpb.OpDef_ArgDef
559 func isListArg(argdef *odpb.OpDef_ArgDef) bool { argument
/external/tensorflow/tensorflow/js/ops/
Dts_op_gen.cc28 static bool IsListAttr(const OpDef_ArgDef& arg) { in IsListAttr()
/external/tensorflow/tensorflow/compiler/tf2xla/
Dtf2xla_util.cc280 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/
Dc_api_experimental.cc456 const tensorflow::OpDef_ArgDef& input_arg = op_def->input_arg()[input_index]; in TF_GetNumberAttrForOpListInput()