| /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 | 64 op: "BiasAdd"
98 ('BiasAdd', 'BiasOp<CPUDevice, float>'), #
114 ('BiasAdd', 'BiasOp<CPUDevice, float>'), #
192 || isequal(op, "BiasAdd")
|
| /external/tensorflow/tensorflow/cc/gradients/ |
| nn_grad_test.cc | 28 using ops::BiasAdd;
167 auto y = BiasAdd(scope_, x, bias);
|
| /external/tensorflow/tensorflow/core/kernels/ |
| cwise_ops_test.cc | 131 Graph* BiasAdd(int rows, int cols, DataType type) {
139 test::graph::Binary(g, "BiasAdd", test::graph::Constant(g, lhs),
151 test::Benchmark(#DEVICE, BiasAdd<C_TYPE>(rows, cols, TF_TYPE)).Run(iters); \
|
| meta_support.cc | 354 gemmlowp::meta::BiasAdd<uint8_t>>
|
| /external/tensorflow/tensorflow/core/graph/ |
| testlib.h | 200 // Add a BiasAdd node in "g".
201 Node* BiasAdd(Graph* g, Node* value, Node* bias);
|
| testlib.cc | 471 Node* BiasAdd(Graph* g, Node* value, Node* bias) {
473 TF_CHECK_OK(NodeBuilder(g->NewName("n"), "BiasAdd")
|
| /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/core/grappler/optimizers/ |
| constant_folding_test.cc | 186 Output bias_add1 = ops::BiasAdd(s.WithOpName("bias_add1"), x, zeros_1d);
187 Output bias_add2 = ops::BiasAdd(s.WithOpName("bias_add2"), zeros, bias);
281 EXPECT_EQ("BiasAdd", node.op());
[all...] |
| /external/tensorflow/tensorflow/compiler/tests/ |
| randomized_tests.cc | [all...] |
| /external/tensorflow/tensorflow/go/op/ |
| wrappers.go | [all...] |
