HomeSort by relevance Sort by last modified time
    Searched defs:OpDefBuilder (Results 1 - 2 of 2) sorted by null

  /external/tensorflow/tensorflow/core/framework/
op_def_builder.h 51 class OpDefBuilder {
54 explicit OpDefBuilder(StringPiece op_name);
56 // Adds an attr to this OpDefBuilder (and returns *this). The spec has
87 OpDefBuilder& Attr(StringPiece spec);
89 // Adds an input or output to this OpDefBuilder (and returns *this).
104 OpDefBuilder& Input(StringPiece spec);
105 OpDefBuilder& Output(StringPiece spec);
107 // Turns on the indicated boolean flag in this OpDefBuilder (and
109 OpDefBuilder& SetIsCommutative();
110 OpDefBuilder& SetIsAggregate()
    [all...]
op_def_builder.cc 524 OpDefBuilder::OpDefBuilder(StringPiece op_name) {
528 OpDefBuilder& OpDefBuilder::Attr(StringPiece spec) {
533 OpDefBuilder& OpDefBuilder::Input(StringPiece spec) {
538 OpDefBuilder& OpDefBuilder::Output(StringPiece spec) {
544 OpDefBuilder& OpDefBuilder::Doc(StringPiece text)
    [all...]

Completed in 63 milliseconds