| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_fused_ops_test.cc | 53 const Tensor& bias_data, const std::vector<string>& fused_ops, Tensor* out,
59 const std::vector<string>& fused_ops, Tensor* out)>;
148 const std::vector<string>& fused_ops, const FusedGraphRunner& run_default, in VerifyFusedTensorsClose() argument
165 run_default(image, filter, bias, fused_ops, &conv_2d, padding); in VerifyFusedTensorsClose()
166 run_fused(image, filter, bias, fused_ops, &fused_conv_2d, padding); in VerifyFusedTensorsClose()
175 const std::vector<string>& fused_ops, in VerifyFusedMatrixClose() argument
192 run_default(input, weight, bias, fused_ops, &output); in VerifyFusedMatrixClose()
193 run_fused(input, weight, bias, fused_ops, &fused_output); in VerifyFusedMatrixClose()
217 const std::vector<string>& fused_ops, Tensor* output, in RunConv2DUnfused() argument
239 if (std::find(fused_ops.begin(), fused_ops.end(), "BiasAdd") != in RunConv2DUnfused()
[all …]
|
| D | mkl_conv_ops.cc | 1168 std::vector<string> fused_ops; in MklFusedConvOp() local
1169 OP_REQUIRES_OK(context, context->GetAttr("fused_ops", &fused_ops)); in MklFusedConvOp()
1173 OP_REQUIRES(context, !fused_ops.empty(), in MklFusedConvOp()
1177 if (fused_ops == std::vector<string>{"BiasAdd"}) { in MklFusedConvOp()
1182 } else if (fused_ops == std::vector<string>{"Relu"}) { in MklFusedConvOp()
1184 } else if (fused_ops == std::vector<string>{"Relu6"}) { in MklFusedConvOp()
1187 } else if (fused_ops == std::vector<string>{"Elu"}) { in MklFusedConvOp()
1189 } else if (fused_ops == std::vector<string>{"LeakyRelu"}) { in MklFusedConvOp()
1195 } else if (fused_ops == std::vector<string>{"BiasAdd", "Relu"}) { in MklFusedConvOp()
1201 } else if (fused_ops == std::vector<string>{"BiasAdd", "Relu6"}) { in MklFusedConvOp()
[all …]
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | fused_eigen_output_kernels.cc | 31 std::vector<string> fused_ops; in InitializeFusedComputation() local
32 TF_RETURN_IF_ERROR(context->GetAttr("fused_ops", &fused_ops)); in InitializeFusedComputation()
33 if (fused_ops.empty()) { in InitializeFusedComputation()
49 if (fused_ops == pattern.fused_ops) { in InitializeFusedComputation()
56 absl::StrJoin(fused_ops, ","), "]"); in InitializeFusedComputation()
|
| D | conv_ops_benchmark_test.cc | 220 const std::vector<string>& fused_ops = {}, in FusedConv2DWithBias() argument
248 .Attr("fused_ops", fused_ops)
259 int out_depth, const std::vector<string>& fused_ops = {}, in FusedConv2DWithBatchNorm() argument
293 .Attr("fused_ops", fused_ops)
|
| D | matmul_op_test.cc | 152 const std::vector<string>& fused_ops, bool transpose_a, in RunFusedMatMulOp() argument
179 .Attr("fused_ops", fused_ops) in RunFusedMatMulOp()
|
| D | fused_eigen_output_kernels.h | 58 std::vector<string> fused_ops; member
|
| D | conv_ops_test.cc | 741 const std::vector<string>& fused_ops, in RunFusedConv2DOp() argument
773 .Attr("fused_ops", fused_ops) in RunFusedConv2DOp()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | mkl_remapper_test.cc | 148 const auto fused_ops = node.attr().at("fused_ops").list().s(); in FuseConv2DWithBiasAndAddNOrAdd() local
150 EXPECT_EQ(3, fused_ops.size()); in FuseConv2DWithBiasAndAddNOrAdd()
151 EXPECT_EQ("BiasAdd", fused_ops[0]); in FuseConv2DWithBiasAndAddNOrAdd()
152 EXPECT_EQ("Add", fused_ops[1]); in FuseConv2DWithBiasAndAddNOrAdd()
153 EXPECT_EQ(activation, fused_ops[2]); in FuseConv2DWithBiasAndAddNOrAdd()
155 EXPECT_EQ(2, fused_ops.size()); in FuseConv2DWithBiasAndAddNOrAdd()
156 EXPECT_EQ("BiasAdd", fused_ops[0]); in FuseConv2DWithBiasAndAddNOrAdd()
157 EXPECT_EQ("Add", fused_ops[1]); in FuseConv2DWithBiasAndAddNOrAdd()
286 const auto fused_ops = node.attr().at("fused_ops").list().s(); \
288 ASSERT_EQ(fused_ops.size(), 1); \
[all …]
|
| D | remapper_test.cc | 378 const auto fused_ops = node.attr().at("fused_ops").list().s(); in TEST_F() local
379 ASSERT_EQ(fused_ops.size(), 1); in TEST_F()
380 EXPECT_EQ(fused_ops[0], "BiasAdd"); in TEST_F()
442 const auto fused_ops = node.attr().at("fused_ops").list().s(); in RunTest() local
443 ASSERT_EQ(fused_ops.size(), 1); in RunTest()
444 EXPECT_EQ(fused_ops[0], "BiasAdd"); in RunTest()
526 const auto fused_ops = node.attr().at("fused_ops").list().s(); in TEST_F() local
527 ASSERT_EQ(fused_ops.size(), 2); in TEST_F()
528 EXPECT_EQ(fused_ops[0], "BiasAdd"); in TEST_F()
529 EXPECT_EQ(fused_ops[1], "Relu"); in TEST_F()
[all …]
|
| D | remapper.cc | 1051 const absl::Span<const absl::string_view> fused_ops, in IsCpuCompatibleDataType() argument
1054 SetAttrValue(fused_ops, &(*attr)["fused_ops"]); in IsCpuCompatibleDataType()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/ |
| D | fused_kernel_matcher.mlir | 9 …s = [1, 1, 1, 1], epsilon = 0.000000e+00 : f32, explicit_paddings = [], fused_ops = ["BiasAdd"], p…
20 …s = [1, 1, 1, 1], epsilon = 0.000000e+00 : f32, explicit_paddings = [], fused_ops = ["BiasAdd", "R…
32 …s = [1, 1, 1, 1], epsilon = 0.000000e+00 : f32, explicit_paddings = [], fused_ops = ["BiasAdd", "R…
44 …s = [1, 1, 1, 1], epsilon = 0.000000e+00 : f32, explicit_paddings = [], fused_ops = ["BiasAdd", "E…
67 …s = [1, 1, 1, 1], epsilon = 0.000000e+00 : f32, explicit_paddings = [], fused_ops = ["BiasAdd"], p…
117 …= "tf._FusedMatMul"(%arg1, %arg2, %arg0) {epsilon = 0.000000e+00 : f32, fused_ops = ["BiasAdd"], t…
128 …= "tf._FusedMatMul"(%arg1, %arg2, %arg0) {epsilon = 0.000000e+00 : f32, fused_ops = ["BiasAdd", "R…
140 …= "tf._FusedMatMul"(%arg1, %arg2, %arg0) {epsilon = 0.000000e+00 : f32, fused_ops = ["BiasAdd", "R…
152 …= "tf._FusedMatMul"(%arg1, %arg2, %arg0) {epsilon = 0.000000e+00 : f32, fused_ops = ["BiasAdd", "E…
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | fused_kernel_matcher.cc | 127 SmallVector<Attribute, 2> fused_ops{StringAttr::get( in matchAndRewrite() local
141 fused_ops.push_back( in matchAndRewrite()
160 ArrayAttr fused_ops_attr = ArrayAttr::get(context, fused_ops); in matchAndRewrite()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | mkl_layout_pass.cc | 1785 std::vector<string> fused_ops; in FusedConv2DRewrite() local
1786 TF_CHECK_OK(GetNodeAttr(n->def(), "fused_ops", &fused_ops)); in FusedConv2DRewrite()
1787 return (fused_ops == std::vector<string>{"BiasAdd"} || in FusedConv2DRewrite()
1788 fused_ops == std::vector<string>{"Relu"} || in FusedConv2DRewrite()
1789 fused_ops == std::vector<string>{"Relu6"} || in FusedConv2DRewrite()
1790 fused_ops == std::vector<string>{"Elu"} || in FusedConv2DRewrite()
1791 fused_ops == std::vector<string>{"BiasAdd", "Relu"} || in FusedConv2DRewrite()
1792 fused_ops == std::vector<string>{"BiasAdd", "Relu6"} || in FusedConv2DRewrite()
1793 fused_ops == std::vector<string>{"BiasAdd", "Elu"} || in FusedConv2DRewrite()
1794 fused_ops == std::vector<string>{"BiasAdd", "Add"} || in FusedConv2DRewrite()
[all …]
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_generated_ops.td | 18334 that follows is specified by the `fused_ops` attribute, which is a list of TF op
18363 DefaultValuedAttr<StrArrayAttr, "{}">:$fused_ops,
18383 that follows is specified by the `fused_ops` attribute, which is a list of TF op
18408 DefaultValuedAttr<StrArrayAttr, "{}">:$fused_ops,
|