Searched refs:orig_op (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | mkl_eager_op_rewrite_test.cc | 57 void CheckRewrite(EagerOperation* orig_op, string expected_op_name) { in CheckRewrite() argument
61 EagerOpRewriteRegistry::PRE_EXECUTION, orig_op, &out_op)); in CheckRewrite()
64 string actual_op_name = orig_op->Name(); in CheckRewrite()
89 auto orig_op = CreateOp(conv_ops[i]); \
90 orig_op->MutableAttrs()->Set("T", T); \
91 orig_op->MutableAttrs()->Set("padding", "VALID"); \
92 CheckRewrite(orig_op.get(), \
103 auto orig_op = CreateOp(conv_ops[i]); \
104 orig_op->MutableAttrs()->Set("T", T); \
105 orig_op->MutableAttrs()->Set("padding", "EXPLICIT"); \
[all …]
|
| D | mkl_eager_op_rewrite.cc | 41 Status Run(EagerOperation* orig_op,
45 static Status SetupNewOp(EagerOperation* orig_op, const string mkl_op_name,
50 static Status CreateGenericMklOp(EagerOperation* orig_op,
60 Status RewriteToMklOp(EagerOperation* orig_op,
127 EagerOperation* orig_op, in Run() argument
129 if (ShouldRewriteOp(orig_op)) { in Run()
130 TF_CHECK_OK(RewriteToMklOp(orig_op, out_op)); in Run()
136 EagerOperation* orig_op, const string mkl_op_name, in SetupNewOp() argument
139 new_mkl_op->reset(new tensorflow::EagerOperation(&orig_op->EagerContext())); in SetupNewOp()
143 int num_inputs = orig_op->Inputs().size(); in SetupNewOp()
[all …]
|
| D | eager_op_rewrite_registry_test.cc | 28 Status Run(EagerOperation* orig_op, in Run() argument
33 new tensorflow::EagerOperation(&orig_op->EagerContext()); in Run()
52 EagerOperation orig_op(ctx); in TEST() local
56 EagerOpRewriteRegistry::PRE_EXECUTION, &orig_op, &out_op)); in TEST()
|
| D | eager_op_rewrite_registry.cc | 39 Phase phase, EagerOperation* orig_op, in RunRewrite() argument
43 TF_RETURN_IF_ERROR(rewrite->Run(orig_op, out_op)); in RunRewrite()
|
| D | eager_op_rewrite_registry.h | 38 virtual Status Run(EagerOperation* orig_op,
67 Status RunRewrite(Phase phase, EagerOperation* orig_op,
|
| /external/tensorflow/tensorflow/compiler/mlir/tfr/integration/ |
| D | node_expansion_pass.cc | 36 Status CompositeOpExpansion::Run(EagerOperation* orig_op, in Run() argument
40 if (orig_op->Device() != kVariantDeviceNull) return Status::OK(); in Run()
41 if (orig_op->is_function()) return Status::OK(); in Run()
55 if (absl::StartsWith(orig_op->op_name(), skip)) return Status::OK(); in Run()
63 const NodeDef& ndef = orig_op->MutableAttrs()->BuildNodeDef(); in Run()
64 auto& ctx = orig_op->EagerContext(); in Run()
66 orig_op->MutableAttrs()->CacheKey(orig_op->DeviceName()); in Run()
81 new_op->Reset(fname.c_str(), orig_op->DeviceName().c_str())); in Run()
82 for (auto input : orig_op->GetInputs()) { in Run()
85 new_op->MutableAttrs()->CopyAttributes(orig_op->Attrs()); in Run()
|
| D | node_expansion_pass.h | 34 Status Run(EagerOperation* orig_op,
|
| /external/e2fsprogs/lib/ext2fs/ |
| D | extent.c | 315 int orig_op, op; in ext2fs_extent_get() local
323 orig_op = op = flags & EXT2_EXTENT_MOVE_MASK; in ext2fs_extent_get()
327 if ((orig_op == EXT2_EXTENT_NEXT) || in ext2fs_extent_get()
328 (orig_op == EXT2_EXTENT_NEXT_LEAF)) { in ext2fs_extent_get()
358 if ((orig_op == EXT2_EXTENT_PREV) || in ext2fs_extent_get()
359 (orig_op == EXT2_EXTENT_PREV_LEAF)) { in ext2fs_extent_get()
389 if (orig_op == EXT2_EXTENT_LAST_LEAF) { in ext2fs_extent_get()
451 if ((orig_op == EXT2_EXTENT_PREV) || in ext2fs_extent_get()
452 (orig_op == EXT2_EXTENT_PREV_LEAF)) in ext2fs_extent_get()
562 if (((orig_op == EXT2_EXTENT_NEXT_LEAF) || in ext2fs_extent_get()
[all …]
|