| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | rewrite_tpu_embedding_ops.cc | 52 LogicalResult GetOp(FuncOp func, OpT* result) { in GetOp() function
65 if (failed(GetOp(func, &recv_op))) return signalPassFailure(); in runOnFunction()
68 if (failed(GetOp(func, &send_op))) return signalPassFailure(); in runOnFunction()
|
| /external/tensorflow/tensorflow/core/util/ |
| D | mkl_util_test.cc | 98 EXPECT_EQ(nullptr, lru_cache.GetOp(std::to_string(k))); in TEST()
104 int* int_ptr = lru_cache.GetOp(std::to_string(k)); in TEST()
114 EXPECT_EQ(nullptr, lru_cache.GetOp(std::to_string(k))); in TEST()
|
| D | mkl_util.h | 1708 T* GetOp(const string& key) { in GetOp() function
1796 MklPrimitive* GetOp(const string& key) { in GetOp() function
1798 return lru_cache.GetOp(key); in GetOp()
1975 return this->GetOp(key); in GetReorder()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | generic_layout_optimizer_transposer.cc | 119 const int start = (node.GetOp() == "Concat") ? 1 : 0; in GetConcatDataFaninPorts()
745 << "' with op '" << node->GetOp() << "' from data format '" in TransposeNode()
760 << "' with op '" << node->GetOp() << "' from data format '" in TransposeNode()
783 << "' with op '" << node->GetOp() << "' from data format '" in TransposeNode()
807 << "' with op '" << node->GetOp() << "' from data format '" in TransposeNode()
830 << "' with op '" << node->GetOp() << "' from data format '" in TransposeNode()
864 << "' with op '" << node->GetOp() << "' from data format '" in TransposeNode()
886 << "' with op '" << node->GetOp() << "' from data format '" in TransposeNode()
906 << "' with op '" << node->GetOp() << "' from data format '" in TransposeNode()
929 << "' with op '" << node->GetOp() << "' from data format '" in TransposeNode()
[all …]
|
| D | implementation_selector.cc | 137 if ("Identity" == fanout.node_view()->GetOp()) { in UpdateForwardIdentityNodeDtype()
347 if (node_view->GetOp() != kDeviceIndexOp) { in SelectDeviceIndex()
357 if (fanout.node_view()->GetOp() != kCaseOp && in SelectDeviceIndex()
358 fanout.node_view()->GetOp() != kStatelessCaseOp) in SelectDeviceIndex()
|
| D | generic_layout_optimizer_test.cc | 443 EXPECT_EQ(input_sizes_node->GetOp(), "DataFormatVecPermute"); in TEST_F()
|
| D | remapper.cc | 553 if (node_view->GetOp() != "FusedBatchNorm" && in IsCpuCompatibleDataType()
|
| /external/pdfium/core/fpdfapi/page/ |
| D | cpdf_psengine.cpp | 143 const PDF_PSOP op = m_Operators[i]->GetOp(); in Execute()
153 if (i == 0 || m_Operators[i - 1]->GetOp() != PSOP_PROC) in Execute()
159 if (i < 2 || m_Operators[i - 1]->GetOp() != PSOP_PROC || in Execute()
160 m_Operators[i - 2]->GetOp() != PSOP_PROC) { in Execute()
|
| D | cpdf_psengine.h | 77 PDF_PSOP GetOp() const { return m_op; } in GetOp() function
|
| D | cpdf_psengine_unittest.cpp | 71 PDF_PSOP new_op = new_psop->GetOp(); in TEST()
|
| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_softmax_op.cc | 182 return this->GetOp(key); in GetSoftmaxFwd()
|
| D | mkl_pooling_ops_common.h | 202 return this->GetOp(key); in GetPoolingFwd()
346 return this->GetOp(key); in GetPoolingBwd()
|
| D | mkl_matmul_ops_common.h | 371 return this->GetOp(key); in GetMklDnnMatMulFwd()
685 return this->GetOp(key); in GetMklMatMul()
|
| D | mkl_quantize_op.cc | 201 return this->GetOp(key); in GetReorder()
|
| D | mkl_slice_op.cc | 302 return this->GetOp(key); in GetReorder()
|
| D | mkl_conv_grad_input_ops.cc | 293 return this->GetOp(key); in GetConvBwdInput()
|
| D | mkl_relu_op.cc | 206 return this->GetOp(key); in GetEltwiseFwd()
418 return this->GetOp(key); in GetEltwiseBwd()
|
| D | mkl_conv_grad_filter_ops.cc | 349 return this->GetOp(key); in GetConvBwdFilter()
|
| D | mkl_fused_batch_norm_op.cc | 367 return this->GetOp(key); in GetBatchNormFwd()
624 return this->GetOp(key); in GetBatchNormBwd()
|
| D | mkl_concat_op.cc | 429 return this->GetOp(key); in GetConcatFwd()
|
| D | mkl_conv_ops.cc | 403 return this->GetOp(key); in GetConvFwd()
|
| /external/tensorflow/tensorflow/core/grappler/utils/ |
| D | graph_view_internal.h | 162 const string& GetOp() const { return node()->op(); } in GetOp() function
419 if (diff->graph_view->GetNode(diff->node_index)->GetOp() == op) { in UpdateOp()
|
| D | graph_view_test.cc | 260 TYPED_TEST(TypedNodeViewTest, GetOp) { in TYPED_TEST() argument
271 EXPECT_EQ(a_node->GetOp(), "op_a"); in TYPED_TEST()
275 EXPECT_EQ(b_node->GetOp(), "op_b"); in TYPED_TEST()
279 EXPECT_EQ(c_node->GetOp(), "op_c"); in TYPED_TEST()
283 EXPECT_EQ(d_node->GetOp(), "op_d"); in TYPED_TEST()
810 EXPECT_EQ(node_view->GetOp(), expected_node_view.GetOp()); in CompareGraphViewWithGraph()
|
| /external/tensorflow/tensorflow/python/eager/ |
| D | pywrap_tfe_src.cc | 88 TFE_Op* GetOp(TFE_Context* ctx, const char* op_or_function_name, in GetOp() function
864 TFE_Op* op = GetOp(ctx, op_name, device_name, out_status); in TFE_Py_ExecuteCancelable()
3627 TFE_Op* op = GetOp(ctx, op_name, op_exec_info.device_name, status);
|