Searched defs:bcast (Results 1 - 25 of 27) sorted by relevance

12

/external/eigen/unsupported/test/
H A Dcxx11_tensor_forced_eval.cpp61 Eigen::array<int, 2> bcast; local
62 bcast[0] = 3;
63 bcast[1] = 1;
65 Tensor<float, 2> output_tensor= (input_tensor - input_tensor.maximum(depth_dim).eval().reshape(dims2d).broadcast(bcast));
/external/tensorflow/tensorflow/core/kernels/
H A Deigen_softmax.h51 Eigen::array<typename internal::traits<Input>::Index, 2> bcast; local
52 bcast[0] = dimensions(input)[0];
53 bcast[1] = 1;
61 Eigen::IndexList<int, Eigen::type2index<1> > bcast;
62 bcast.set(0, dimensions(input)[0]);
71 input.maximum(depth_dim).eval().reshape(dims2d).broadcast(bcast)) *
76 (output.sum(depth_dim).eval().reshape(dims2d).broadcast(bcast));
H A Deigen_softmax_test.cc41 Eigen::array<int, 2> bcast; local
42 bcast[0] = depth;
43 bcast[1] = 1;
49 input.maximum(depth_dim).eval().reshape(dims2d).broadcast(bcast)) *
54 (reference.sum(depth_dim).eval().reshape(dims2d).broadcast(bcast));
H A Dops_util_test.cc124 static void VerifyBoundaries(bcast_struct bcast, error::Code code) { argument
127 bcast.input.index, bcast.input.in_size, bcast.input.ksize,
128 bcast.input.stride, bcast.input.pad_size, &new_index, &new_size);
132 static void VerifyBcastValues(bcast_struct bcast) { argument
135 GetBroadcastSize(bcast.input.index, bcast.input.in_size,
136 bcast
178 bcast_struct bcast = {{2, 3, 1, 2, 0}, {0, 3}}; local
184 bcast_struct bcast[] = { local
196 bcast_struct bcast[] = { local
208 bcast_struct bcast[] = { local
220 bcast_struct bcast[] = { local
231 bcast_struct bcast[] = { local
242 bcast_struct bcast[] = { local
252 bcast_struct bcast[] = { local
262 bcast_struct bcast[] = { local
273 bcast_struct bcast[] = { local
283 bcast_struct bcast[] = { local
294 bcast_struct bcast[] = { local
304 bcast_struct bcast[] = { local
[all...]
H A Dtraining_ops_gpu.cu.cc33 Eigen::array<typename TTypes<T>::Tensor::Index, 1> bcast; local
34 bcast[0] = grad.dimension(0);
36 var.device(d) -= lr.reshape(single).broadcast(bcast) * grad;
47 Eigen::array<typename TTypes<T>::Tensor::Index, 1> bcast; local
48 bcast[0] = grad.dimension(0);
50 var.device(d) -= lr.reshape(single).broadcast(bcast) * grad * accum.rsqrt();
63 Eigen::array<typename TTypes<T>::Tensor::Index, 1> bcast; local
64 bcast[0] = grad.dimension(0);
67 accum.device(d) = accum * rho.reshape(single).broadcast(bcast) +
69 rho.reshape(single).broadcast(bcast));
88 Eigen::array<typename TTypes<T>::Tensor::Index, 1> bcast; local
113 Eigen::array<typename TTypes<T>::Tensor::Index, 1> bcast; local
154 Eigen::array<typename TTypes<T>::Tensor::Index, 1> bcast; local
179 Eigen::array<typename TTypes<T>::Tensor::Index, 1> bcast; local
203 Eigen::array<typename TTypes<T>::Tensor::Index, 1> bcast; local
235 Eigen::array<typename TTypes<T>::Tensor::Index, 1> bcast; local
[all...]
H A Dpooling_ops_3d.cc258 Eigen::array<int, 5> bcast = {1, csize, rsize, psize, 1}; local
262 bcast; local
263 bcast.set(1, csize);
264 bcast.set(2, rsize);
265 bcast.set(3, psize);
289 ((tensor_in_slice - tensor_out_slice.broadcast(bcast)).abs() <
291 .select(out_backprop_slice.broadcast(bcast), mat0);
431 Eigen::array<int, 5> bcast = {1, csize, rsize, psize, 1}; local
435 bcast; local
436 bcast
[all...]
H A Dcwise_ops_common.h35 #include "tensorflow/core/util/bcast.h"
51 // Sets up bcast with the shape of in0 and in1, ensures that the bcast
62 BCast bcast; member in struct:tensorflow::BinaryOpShared::BinaryOpState
94 BCast* bcast = &state.bcast; variable
123 eigen_device, out->shaped<Tout, 2>(bcast->result_shape()),
124 in0.template shaped<Tin, 2>(bcast->x_reshape()),
125 BCast::ToIndexArray<2>(bcast->x_bcast()),
126 in1.template shaped<Tin, 2>(bcast
[all...]
/external/tensorflow/tensorflow/compiler/xla/service/
H A Dhlo_rematerialization_test.cc44 // F32[1024] %bcast = broadcast(%param)
45 // F32[1024] %negate = negate(%bcast)
48 // F32[1025] %concat_2 = concat({%bcast, %slice_1})
51 // The instruction %bcast can be rematerialized before its use at %concat_2
52 // to reduce peak memory usage. This avoids %bcast and %concat_1 being
54 // (during execution of %concat_1) and about 12KB after rematerializing %bcast
61 auto bcast = builder.AddInstruction( local
64 HloInstruction::CreateUnary(vec1024_shape_, HloOpcode::kNegate, bcast));
73 ShapeUtil::MakeShape(xla::F32, {1025}), {bcast, slice_1},
88 // F32[1024] %bcast
104 auto bcast = builder.AddInstruction( local
156 const HloInstruction* bcast = concat->operand(0); local
437 auto bcast = builder.AddInstruction( local
553 auto bcast = builder.AddInstruction( local
[all...]
H A Dcopy_insertion_test.cc827 auto bcast = body->root_instruction()->operand(1)->operand(1); local
829 ASSERT_EQ(bcast->opcode(), HloOpcode::kBroadcast);
H A Dalgebraic_simplifier_test.cc128 HloInstruction* bcast = builder.AddInstruction( local
131 HloInstruction::CreateBinary(r2f32, HloOpcode::kAdd, bcast, param0));
150 HloInstruction* bcast = local
153 HloInstruction::CreateBinary(r2f32, HloOpcode::kAdd, bcast, param0));
/external/tensorflow/tensorflow/compiler/xla/tests/
H A Dconcat_test.cc552 auto bcast = builder.Broadcast(y, {5}); local
554 auto concat = builder.ConcatInDim({bcast, x}, /*dimension=*/0);
/external/wpa_supplicant_8/hostapd/src/wps/
H A Dwps.c319 const u8 bcast[ETH_ALEN] = { 0xff, 0xff, 0xff, 0xff, 0xff, 0xff }; local
339 if (os_memcmp(pos, bcast, ETH_ALEN) == 0)
H A Dwps_er.c1509 u8 bcast[ETH_ALEN]; local
1524 os_memset(bcast, 0xff, ETH_ALEN);
1525 auth_macs = bcast;
H A Dwps_registrar.c794 u8 bcast[ETH_ALEN] = { 0xff, 0xff, 0xff, 0xff, 0xff, 0xff }; local
797 addr = bcast;
/external/wpa_supplicant_8/src/wps/
H A Dwps.c319 const u8 bcast[ETH_ALEN] = { 0xff, 0xff, 0xff, 0xff, 0xff, 0xff }; local
339 if (os_memcmp(pos, bcast, ETH_ALEN) == 0)
H A Dwps_er.c1509 u8 bcast[ETH_ALEN]; local
1524 os_memset(bcast, 0xff, ETH_ALEN);
1525 auth_macs = bcast;
H A Dwps_registrar.c794 u8 bcast[ETH_ALEN] = { 0xff, 0xff, 0xff, 0xff, 0xff, 0xff }; local
797 addr = bcast;
/external/wpa_supplicant_8/wpa_supplicant/src/wps/
H A Dwps.c319 const u8 bcast[ETH_ALEN] = { 0xff, 0xff, 0xff, 0xff, 0xff, 0xff }; local
339 if (os_memcmp(pos, bcast, ETH_ALEN) == 0)
H A Dwps_er.c1509 u8 bcast[ETH_ALEN]; local
1524 os_memset(bcast, 0xff, ETH_ALEN);
1525 auth_macs = bcast;
H A Dwps_registrar.c794 u8 bcast[ETH_ALEN] = { 0xff, 0xff, 0xff, 0xff, 0xff, 0xff }; local
797 addr = bcast;
/external/libnl/lib/route/
H A Daddr.c977 int rtnl_addr_set_broadcast(struct rtnl_addr *addr, struct nl_addr *bcast) argument
979 if (bcast && bcast->a_family != AF_INET)
982 return __assign_addr(addr, &addr->a_bcast, bcast, ADDR_ATTR_BROADCAST);
/external/spirv-llvm/lib/SPIRV/libSPIRV/
H A DSPIRVEntry.h679 T* bcast(SPIRVEntry *E) { function in namespace:SPIRV
/external/tensorflow/tensorflow/contrib/tensor_forest/kernels/
H A Dtree_utils.cc100 Eigen::array<int, 1> bcast; local
101 bcast[0] = num_splits;
103 tc.broadcast(bcast) - splits;
181 Eigen::array<int, 1> bcast; local
182 bcast[0] = num_splits;
183 const auto right_sums = tc_sum.broadcast(bcast) - splits_sum;
184 const auto right_squares = tc_square.broadcast(bcast) - splits_square;
/external/toybox/toys/pending/
H A Dip.c61 iface[IFNAMSIZ+1], laddr[64], bcast[64]; member in struct:linkdata
783 line_feed, link->type, link->laddr, peer ,link->bcast);
858 inet_ntop(AF_INET, RTA_DATA(attr), link->bcast, 64);
859 else fill_address(RTA_DATA(attr), link->bcast);
/external/wpa_supplicant_8/hostapd/src/drivers/
H A Ddriver_ndis.c1042 u8 bcast[ETH_ALEN] = { 0xff, 0xff, 0xff, 0xff, 0xff, 0xff }; local
1075 bcast, i,
1113 bcast, 0, 1,

Completed in 1929 milliseconds

12