Searched defs:fused_ops (Results 1 – 12 of 12) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_fused_ops_test.cc | 148 const std::vector<string>& fused_ops, const FusedGraphRunner& run_default, in VerifyFusedTensorsClose()
175 const std::vector<string>& fused_ops, in VerifyFusedMatrixClose()
217 const std::vector<string>& fused_ops, Tensor* output, in RunConv2DUnfused()
283 const std::vector<string>& fused_ops, Tensor* output, in RunMklFusedConv2DOp()
344 const std::vector<string>& fused_ops, in VerifyFusedConv2D()
359 const Tensor& bias_data, const std::vector<string>& fused_ops, in VerifyFusedConv2D()
360 Tensor* out, const int padding) { in VerifyFusedConv2D()
557 const std::vector<string>& fused_ops, in RunDepthwiseConv2DUnfused()
599 const std::vector<string>& fused_ops, in RunMklFusedDepthwiseConv2DOp()
667 const std::vector<string>& fused_ops, in VerifyFusedDepthwiseConv2D()
[all …]
|
| D | mkl_conv_ops.cc | 1168 std::vector<string> fused_ops; in MklFusedConvOp() local
1299 std::vector<string> fused_ops; in MklFusedDepthwiseConvOp() local
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | fused_eigen_output_kernels.cc | 31 std::vector<string> fused_ops; in InitializeFusedComputation() local
|
| D | fused_eigen_output_kernels.h | 58 std::vector<string> fused_ops; member
|
| D | conv_ops_benchmark_test.cc | 220 const std::vector<string>& fused_ops = {}, in FusedConv2DWithBias()
259 int out_depth, const std::vector<string>& fused_ops = {}, in FusedConv2DWithBatchNorm()
|
| D | matmul_op_test.cc | 152 const std::vector<string>& fused_ops, bool transpose_a, in RunFusedMatMulOp()
|
| D | conv_ops_test.cc | 741 const std::vector<string>& fused_ops, in RunFusedConv2DOp()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | remapper_test.cc | 378 const auto fused_ops = node.attr().at("fused_ops").list().s(); in TEST_F() local
442 const auto fused_ops = node.attr().at("fused_ops").list().s(); in RunTest() local
526 const auto fused_ops = node.attr().at("fused_ops").list().s(); in TEST_F() local
613 const auto fused_ops = node.attr().at("fused_ops").list().s(); in TEST_F() local
715 const auto fused_ops = node.attr().at("fused_ops").list().s(); in RunTest() local
816 const auto fused_ops = node.attr().at("fused_ops").list().s(); in TEST_F() local
915 const auto fused_ops = node.attr().at("fused_ops").list().s(); in TEST_F() local
988 const auto fused_ops = node.attr().at("fused_ops").list().s(); in TEST_F() local
|
| D | mkl_remapper_test.cc | 148 const auto fused_ops = node.attr().at("fused_ops").list().s(); in FuseConv2DWithBiasAndAddNOrAdd() local
512 const auto fused_ops = node.attr().at("fused_ops").list().s(); in TEST_F() local
|
| D | remapper.cc | 1051 const absl::Span<const absl::string_view> fused_ops, in IsCpuCompatibleDataType()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | fused_kernel_matcher.cc | 127 SmallVector<Attribute, 2> fused_ops{StringAttr::get( in matchAndRewrite() local
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | mkl_layout_pass.cc | 1785 std::vector<string> fused_ops; in FusedConv2DRewrite() local
1814 std::vector<string> fused_ops; in FusedDepthwiseConv2DRewrite() local
2710 std::vector<string> fused_ops; in CopyAttrsFromPadAndFusedConv2D() local
|