| /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()
|