Searched refs:TapeOperation (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/c/experimental/gradients/tape/ |
| D | tape_operation.cc | 22 TapeOperation::TapeOperation(AbstractOperation* parent_op, Tape* tape, in TapeOperation() function in tensorflow::gradients::TapeOperation
31 void TapeOperation::Release() { in Release()
35 TapeOperation::~TapeOperation() { in ~TapeOperation()
39 Status TapeOperation::Reset(const char* op, const char* raw_device_name) { in Reset()
46 const string& TapeOperation::Name() const { return parent_op_->Name(); } in Name()
47 const string& TapeOperation::DeviceName() const { in DeviceName()
50 Status TapeOperation::SetDeviceName(const char* name) { in SetDeviceName()
53 Status TapeOperation::AddInput(AbstractTensorHandle* input) { in AddInput()
58 Status TapeOperation::AddInputList( in AddInputList()
66 Status TapeOperation::SetAttrString(const char* attr_name, const char* data, in SetAttrString()
[all …]
|
| D | tape_operation.h | 23 class TapeOperation : public AbstractOperation {
25 explicit TapeOperation(AbstractOperation*, Tape*, const GradientRegistry&);
69 ~TapeOperation() override;
|
| D | tape_context.cc | 36 TapeOperation* TapeContext::CreateOperation() { in CreateOperation()
37 return new TapeOperation(parent_ctx_->CreateOperation(), tape_, registry_); in CreateOperation()
|
| D | tape_context.h | 27 TapeOperation* CreateOperation() override;
|
| /external/tensorflow/tensorflow/c/eager/ |
| D | tracing_utils.cc | 29 if (isa<gradients::TapeOperation>(op)) { in MaybeSetOpName()
31 dyn_cast<gradients::TapeOperation>(op)->GetBackingOperation(), in MaybeSetOpName()
|