| /external/gemmlowp/meta/ |
| transform_kernels_arm_32.h | [all...] |
| transform_kernels_arm_64.h | [all...] |
| transform_kernels.h | 55 struct BiasAdd {
145 class Transform1DKernel<InType, OutType, BiasAdd<Type>, kernel_size,
148 static void Transform(const InType* in, const BiasAdd<Type>& params,
152 std::cout << "BiasAdd::Transform(" << std::string(typeid(InType).name())
157 std::cerr << "FATAL: BiasAdd::Transform not implemented." << std::endl;
|
| test_transform_correctness.cc | 40 typedef Transform1DParams<std::uint8_t, std::int32_t, BiasAdd<std::uint8_t>> BiasAddParams;
170 std::cout << "BiasAdd: OK" << std::endl;
|
| /external/tensorflow/tensorflow/python/tools/ |
| print_selective_registration_header_test.py | 68 op: "BiasAdd"
113 ('BiasAdd', 'BiasOp<CPUDevice, float>'), #
131 ('BiasAdd', 'BiasOp<CPUDevice, float>'), #
211 || isequal(op, "BiasAdd")
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| remapper_test.cc | 121 auto bias_add = ops::BiasAdd(s.WithOpName("bias_add"), conv, bias);
154 EXPECT_EQ("BiasAdd", fused_ops[0]);
184 auto bias_add = ops::BiasAdd(s.WithOpName("bias_add"), conv, bias);
218 EXPECT_EQ("BiasAdd", fused_ops[0]);
410 auto bias_add = ops::BiasAdd(s.WithOpName("bias_add"), squeeze, bias);
443 EXPECT_EQ("BiasAdd", fused_ops[0]);
|
| constant_folding_test.cc | 526 Output bias_add1 = ops::BiasAdd(s.WithOpName("bias_add1"), x, zeros_1d);
527 Output bias_add2 = ops::BiasAdd(s.WithOpName("bias_add2"), zeros, bias);
628 EXPECT_EQ("BiasAdd", node.op());
[all...] |
| /external/tensorflow/tensorflow/cc/gradients/ |
| nn_grad_test.cc | 31 using ops::BiasAdd;
223 auto y = BiasAdd(scope_, x, bias);
|
| /external/tensorflow/tensorflow/core/graph/ |
| testlib.h | 204 // Add a BiasAdd node in "g".
205 Node* BiasAdd(Graph* g, Node* value, Node* bias);
|
| testlib.cc | 459 Node* BiasAdd(Graph* g, Node* value, Node* bias) {
461 TF_CHECK_OK(NodeBuilder(g->NewName("n"), "BiasAdd")
|
| /external/tensorflow/tensorflow/core/kernels/ |
| mkl_fused_ops_test.cc | 143 auto with_bias = ops::BiasAdd(
162 auto with_bias = ops::BiasAdd(
215 // Verifies that computing Conv2D+BiasAdd in a graph is identical to
230 RunMklFusedConv2DOp(input_data, filter_data, {bias_data}, {"BiasAdd"},
239 // Verifies that computing Conv2D+BiasAdd+Relu in a graph is identical to
256 {"BiasAdd", "Relu"}, out);
271 // Conv2D + BiasAdd + {Relu} //
486 // Verifies that computing Pad+Conv2D+BiasAdd in a graph is identical to
504 {"BiasAdd"}, out);
512 // Verifies that computing Pad+Conv2D+BiasAdd+Relu in a graph is identical t
[all...] |
| cwise_ops_test.cc | 142 Graph* BiasAdd(int rows, int cols, DataType type) {
150 test::graph::Binary(g, "BiasAdd", test::graph::Constant(g, lhs),
163 test::Benchmark(#DEVICE, BiasAdd<C_TYPE>(rows, cols, TF_TYPE)).Run(iters); \
|
| conv_ops_test.cc | 620 ops::BiasAdd with_bias = ops::BiasAdd(
639 ops::BiasAdd with_bias = ops::BiasAdd(
[all...] |
| meta_support.cc | 354 gemmlowp::meta::BiasAdd<uint8_t>>
|
| /external/tensorflow/tensorflow/tools/graph_transforms/ |
| quantize_nodes_test.cc | 109 "Add", "BiasAdd", "Concat", "Conv2D", "MatMul", "Relu",
359 BiasAdd(root.WithOpName("bias_add_op"), input_op, offset_op);
[all...] |
| /external/tensorflow/tensorflow/cc/framework/ |
| cc_ops_test.cc | 31 return BiasAdd(cop_scopes.last, m, b);
|
| /external/gemmlowp/meta/generators/ |
| transform_kernels_common.py | 391 class BiasAdd(common.Transform1DKernelGenerator):
396 'BiasAdd<%s>' % bias_type)
490 emitter.EmitComment('BiasAdd::Transform')
572 biasadd = BiasAdd('uint8_t', cc_emitter, asm_emitter)
589 biasadd.SpecializeTransform1DKernel('uint8_t', 'int32_t', shape[0],
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| convert_nodes_test.cc | [all...] |
| /external/tensorflow/tensorflow/compiler/tests/ |
| randomized_tests.cc | [all...] |
| /external/tensorflow/tensorflow/go/op/ |
| wrappers.go | [all...] |
