/external/tensorflow/tensorflow/c/eager/ |
D | gradients_internal.h | 30 Status Reset(AbstractOperation*, const char* op, const char* raw_device_name, 34 Status AddInput(AbstractOperation*, AbstractTensorHandle*, ForwardOperation*); 35 Status AddInputList(AbstractOperation*, 40 Status SetAttrString(AbstractOperation*, const char* attr_name, 42 Status SetAttrInt(AbstractOperation*, const char* attr_name, int64_t value, 44 Status SetAttrFloat(AbstractOperation*, const char* attr_name, float value, 46 Status SetAttrBool(AbstractOperation*, const char* attr_name, bool value, 48 Status SetAttrType(AbstractOperation*, const char* attr_name, DataType value, 50 Status SetAttrShape(AbstractOperation*, const char* attr_name, 52 Status SetAttrFunction(AbstractOperation*, const char* attr_name, [all …]
|
D | gradients.cc | 301 Status Reset(AbstractOperation* op_, const char* op, in Reset() 307 Status AddInput(AbstractOperation* op_, AbstractTensorHandle* input, in AddInput() 313 Status AddInputList(AbstractOperation* op_, in AddInputList() 323 Status SetAttrString(AbstractOperation* op_, const char* attr_name, in SetAttrString() 329 Status SetAttrInt(AbstractOperation* op_, const char* attr_name, int64_t value, in SetAttrInt() 334 Status SetAttrFloat(AbstractOperation* op_, const char* attr_name, float value, in SetAttrFloat() 339 Status SetAttrBool(AbstractOperation* op_, const char* attr_name, bool value, in SetAttrBool() 344 Status SetAttrType(AbstractOperation* op_, const char* attr_name, in SetAttrType() 349 Status SetAttrShape(AbstractOperation* op_, const char* attr_name, in SetAttrShape() 370 Status SetAttrFunction(AbstractOperation* op_, const char* attr_name, in SetAttrFunction() [all …]
|
D | abstract_operation.h | 31 class AbstractOperation { 41 explicit AbstractOperation(AbstractOperationKind kind) : kind_(kind) {} in AbstractOperation() function 42 virtual ~AbstractOperation() {} in ~AbstractOperation() 97 const AbstractOperation* value) = 0; 117 const char* attr_name, absl::Span<const AbstractOperation*> values) = 0; 125 void operator()(AbstractOperation* p) const { in operator() 134 std::unique_ptr<AbstractOperation, internal::AbstractOperationDeleter>;
|
D | c_api_unified_experimental_internal.h | 73 class TracingOperation : public AbstractOperation { 76 : AbstractOperation(kind) {} in TracingOperation() 85 static bool classof(const AbstractOperation* ptr) { in classof() 133 DEFINE_CONVERSION_FUNCTIONS(AbstractOperation, TF_AbstractOp)
|
D | immediate_execution_operation.h | 40 class ImmediateExecutionOperation : public AbstractOperation { 72 static bool classof(const AbstractOperation* ptr) { in classof() 78 : AbstractOperation(kind) {} in ImmediateExecutionOperation()
|
D | tracing_utils.h | 22 Status MaybeSetOpName(AbstractOperation*, const char* op_name);
|
D | tracing_utils.cc | 25 Status MaybeSetOpName(AbstractOperation* op, const char* op_name) { in MaybeSetOpName()
|
D | abstract_context.h | 54 virtual AbstractOperation* CreateOperation() = 0;
|
D | c_api_unified_experimental_graph.cc | 215 const AbstractOperation* value) override { in SetAttrFunction() 301 absl::Span<const AbstractOperation*> values) override { in SetAttrFunctionList() 306 static bool classof(const AbstractOperation* ptr) { in classof()
|
/external/tensorflow/tensorflow/c/experimental/gradients/tape/ |
D | tape_operation.h | 23 class TapeOperation : public AbstractOperation { 25 explicit TapeOperation(AbstractOperation*, Tape*, const GradientRegistry&); 44 const AbstractOperation* value) override; 63 absl::Span<const AbstractOperation*> values) override; 64 AbstractOperation* GetBackingOperation(); 66 static bool classof(const AbstractOperation* ptr) { in classof() 72 AbstractOperation* parent_op_;
|
D | tape_operation.cc | 22 TapeOperation::TapeOperation(AbstractOperation* parent_op, Tape* tape, in TapeOperation() 24 : AbstractOperation(kTape), in TapeOperation() 108 const AbstractOperation* value) { in SetAttrFunction() 191 const char* attr_name, absl::Span<const AbstractOperation*> values) { in SetAttrFunctionList() 196 AbstractOperation* TapeOperation::GetBackingOperation() { return parent_op_; } in GetBackingOperation()
|
/external/tensorflow/tensorflow/python/framework/experimental/ |
D | unified_api.cc | 48 using tensorflow::AbstractOperation; 184 py::class_<AbstractOperation, AbstractOperationPtr>(m, "AbstractOperation") in PYBIND11_MODULE() 186 [](AbstractOperation* self, const char* op, in PYBIND11_MODULE() 192 [](AbstractOperation* self, const char* op_name) { in PYBIND11_MODULE() 201 .def("Name", &AbstractOperation::Name) in PYBIND11_MODULE() 202 .def("DeviceName", &AbstractOperation::DeviceName) in PYBIND11_MODULE() 204 [](AbstractOperation* self, const char* name) { in PYBIND11_MODULE() 209 [](AbstractOperation* self, AbstractTensorHandle* input) { in PYBIND11_MODULE() 214 [](AbstractOperation* self, const char* attr_name, DataType value) { in PYBIND11_MODULE() 218 .def("Execute", [](AbstractOperation* self, int num_outputs) { in PYBIND11_MODULE()
|
/external/tensorflow/tensorflow/c/experimental/op_handler/ |
D | wrapper_operation.h | 26 class WrapperOperation : public AbstractOperation { 28 explicit WrapperOperation(AbstractOperation*, AbstractOperationKind kind); 47 const AbstractOperation* value) override; 66 absl::Span<const AbstractOperation*> values) override; 67 AbstractOperation* GetBackingOperation(); 70 AbstractOperation* parent_op_;
|
D | wrapper_operation.cc | 18 WrapperOperation::WrapperOperation(AbstractOperation* parent_op, in WrapperOperation() 20 : AbstractOperation(kind), parent_op_(parent_op) { in WrapperOperation() 68 const AbstractOperation* value) { in SetAttrFunction() 109 const char* attr_name, absl::Span<const AbstractOperation*> values) { in SetAttrFunctionList() 112 AbstractOperation* WrapperOperation::GetBackingOperation() { in GetBackingOperation()
|
D | internal.cc | 56 OpHandlerOperation::OpHandlerOperation(AbstractOperation* parent_op) in OpHandlerOperation()
|
D | internal.h | 68 explicit OpHandlerOperation(AbstractOperation*);
|
/external/llvm-project/mlir/lib/IR/ |
D | MLIRContext.cpp | 263 llvm::StringMap<AbstractOperation> registeredOperations; 552 std::vector<AbstractOperation *> MLIRContext::getRegisteredOperations() { in getRegisteredOperations() 556 llvm::StringMap<AbstractOperation> ®isteredOps = in getRegisteredOperations() 559 std::vector<AbstractOperation *> result; in getRegisteredOperations() 565 [](AbstractOperation *const *lhs, AbstractOperation *const *rhs) { in getRegisteredOperations() 619 ParseResult AbstractOperation::parseAssembly(OpAsmParser &parser, in parseAssembly() 626 const AbstractOperation *AbstractOperation::lookup(StringRef opName, in lookup() 635 void AbstractOperation::insert( in insert() 641 AbstractOperation opInfo(name, dialect, opProperties, typeID, parseAssembly, in insert() 657 AbstractOperation::AbstractOperation( in AbstractOperation() function in AbstractOperation
|
/external/llvm-project/mlir/include/mlir/IR/ |
D | OpDefinition.h | 260 friend AbstractOperation; variable 689 static AbstractOperation::OperationProperties getTraitProperties() { in getTraitProperties() 690 return static_cast<AbstractOperation::OperationProperties>( in getTraitProperties() 1021 static AbstractOperation::OperationProperties getTraitProperties() { in getTraitProperties() 1022 return static_cast<AbstractOperation::OperationProperties>( in getTraitProperties() 1128 static AbstractOperation::OperationProperties getTraitProperties() { in getTraitProperties() 1129 return static_cast<AbstractOperation::OperationProperties>( in getTraitProperties() 1456 static AbstractOperation::OperationProperties 1458 AbstractOperation::OperationProperties result = 0; in getTraitPropertiesImpl() 1467 static AbstractOperation::OperationProperties getTraitProperties() { in getTraitProperties() [all …]
|
D | MLIRContext.h | 19 class AbstractOperation; variable 130 std::vector<AbstractOperation *> getRegisteredOperations();
|
D | OperationSupport.h | 82 class AbstractOperation { 169 static const AbstractOperation *lookup(StringRef opName, 191 AbstractOperation(StringRef name, Dialect &dialect, 328 PointerUnion<Identifier, const AbstractOperation *>; 330 OperationName(AbstractOperation *op) : representation(op) {} in OperationName() 347 const AbstractOperation *getAbstractOperation() const;
|
/external/tensorflow/tensorflow/java/src/main/java/org/tensorflow/ |
D | Output.java | 97 Output(AbstractOperation op, int idx) { in Output() 106 private final AbstractOperation operation;
|
D | AbstractOperation.java | 24 abstract class AbstractOperation implements Operation { class
|
D | GraphOperation.java | 27 public final class GraphOperation extends AbstractOperation {
|
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/c/ |
D | c_api_unified_experimental_mlir.cc | 64 using tensorflow::AbstractOperation; 163 const AbstractOperation* value) override; 183 absl::Span<const AbstractOperation*> values) override; 194 static bool classof(const AbstractOperation* ptr) { in classof() 258 AbstractOperation* CreateOperation() override { in CreateOperation() 474 const AbstractOperation* value) { in SetAttrFunction() 515 const char* attr_name, absl::Span<const AbstractOperation*> values) { in SetAttrFunctionList()
|
/external/tensorflow/tensorflow/core/common_runtime/eager/ |
D | eager_operation.h | 106 const AbstractOperation* value) override; 125 absl::Span<const AbstractOperation*> values) override; 196 static bool classof(const AbstractOperation* ptr) { in classof()
|