Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/c/experimental/gradients/tape/
Dtape_operation.cc40 forward_op_.op_name = op; in Reset()
41 forward_op_.attrs.Reset(op); in Reset()
42 forward_op_.inputs.clear(); in Reset()
43 forward_op_.outputs.clear(); in Reset()
55 forward_op_.inputs.push_back(input); in AddInput()
62 forward_op_.inputs.push_back(input); in AddInputList()
68 forward_op_.attrs.Set(attr_name, StringPiece(data, length)); in SetAttrString()
72 forward_op_.attrs.Set(attr_name, static_cast<int64>(value)); in SetAttrInt()
76 forward_op_.attrs.Set(attr_name, value); in SetAttrFloat()
80 forward_op_.attrs.Set(attr_name, value); in SetAttrBool()
[all …]
Dtape_operation.h73 ForwardOperation forward_op_; variable
/external/tensorflow/tensorflow/c/eager/
Dgradients.cc302 const char* raw_device_name, ForwardOperation* forward_op_) { in Reset() argument
303 forward_op_->op_name = op; in Reset()
304 forward_op_->attrs.Reset(op); in Reset()
308 ForwardOperation* forward_op_) { in AddInput() argument
310 forward_op_->inputs.push_back(input); in AddInput()
315 ForwardOperation* forward_op_) { in AddInputList() argument
318 forward_op_->inputs.push_back(input); in AddInputList()
325 ForwardOperation* forward_op_) { in SetAttrString() argument
326 forward_op_->attrs.Set(attr_name, StringPiece(data, length)); in SetAttrString()
330 ForwardOperation* forward_op_) { in SetAttrInt() argument
[all …]