Home
last modified time | relevance | path

Searched refs:ArgumentSpec (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/java/src/gen/cc/
Dop_specs.h56 class ArgumentSpec {
65 ArgumentSpec(const string& op_def_name, const Variable& var, const Type& type, in ArgumentSpec() function
147 const std::vector<ArgumentSpec>& inputs() const { return inputs_; } in inputs()
148 const std::vector<ArgumentSpec>& outputs() const { return outputs_; } in outputs()
170 std::vector<ArgumentSpec> inputs_;
171 std::vector<ArgumentSpec> outputs_;
Dop_generator.cc89 for (const ArgumentSpec& input : op.inputs()) { in CollectOpDependencies()
95 for (const ArgumentSpec& output : op.outputs()) { in CollectOpDependencies()
170 for (const ArgumentSpec& input : op.inputs()) { in RenderSecondaryFactoryMethod()
209 for (const ArgumentSpec& input : op.inputs()) { in RenderFactoryMethods()
235 for (const ArgumentSpec& input : op.inputs()) { in RenderFactoryMethods()
281 for (const ArgumentSpec& output : op.outputs()) { in RenderConstructor()
293 for (const ArgumentSpec& output : op.outputs()) { in RenderConstructor()
331 for (const ArgumentSpec& output : op.outputs()) { in RenderGettersAndSetters()
343 ArgumentSpec output = op.outputs().front(); in RenderInterfaceImpl()
430 const ArgumentSpec& output = op.outputs().front(); in GenerateOp()
[all …]
Dop_specs.cc285 ArgumentSpec CreateInput(const OpDef_ArgDef& input_def, in CreateInput()
295 return ArgumentSpec( in CreateInput()
320 ArgumentSpec CreateOutput(const OpDef_ArgDef& output_def, in CreateOutput()
329 return ArgumentSpec( in CreateOutput()