Home
last modified time | relevance | path

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

/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/core/grappler/optimizers/data/
Dfunction_utils.cc141 [&name](const OpDef_ArgDef& arg) { return arg.name() == name; }, in FindFunctionInputWithName()
147 [&name](const OpDef_ArgDef& arg) { return arg.name() == name; }, in FindFunctionOutputWithName()
Dhoist_random_uniform.cc133 OpDef_ArgDef* InsertSeedArgument(OpDef* signature, int num_placeholders) { in InsertSeedArgument()
Dfusion_utils.cc261 const protobuf::RepeatedPtrField<OpDef_ArgDef>& fake_args, in CreateFalsePredicate()
/external/tensorflow/tensorflow/go/genop/internal/
Dgenop.go372 op *pb.OpDef_ArgDef
555 func isListArg(argdef *pb.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.cc281 const OpDef_ArgDef& output_arg = body_func->signature().output_arg(i); in PropagateConstIntoWhileNode()
288 const OpDef_ArgDef& input_arg = body_func->signature().input_arg(i); in PropagateConstIntoWhileNode()
/external/tensorflow/tensorflow/c/
Dc_api_experimental.cc8874 const tensorflow::OpDef_ArgDef& input_arg = op_def->input_arg()[input_index]; in TF_GetNumberAttrForOpListInput()