Home
last modified time | relevance | path

Searched defs:bcast (Results 1 – 25 of 34) sorted by relevance

12

/external/tensorflow/tensorflow/core/kernels/
Dtraining_ops_gpu.cu.cc33 Eigen::array<typename TTypes<T>::Tensor::Index, 1> bcast; in operator ()() local
49 Eigen::array<typename TTypes<T>::Tensor::Index, 1> bcast; in operator ()() local
65 Eigen::array<typename TTypes<T>::Tensor::Index, 1> bcast; in operator ()() local
90 Eigen::array<typename TTypes<T>::Tensor::Index, 1> bcast; in operator ()() local
111 Eigen::array<typename TTypes<T>::Tensor::Index, 1> bcast; in operator ()() local
136 Eigen::array<typename TTypes<T>::Tensor::Index, 1> bcast; in operator ()() local
180 Eigen::array<typename TTypes<T>::Tensor::Index, 1> bcast; in operator ()() local
211 Eigen::array<typename TTypes<T>::Tensor::Index, 1> bcast; in operator ()() local
236 Eigen::array<typename TTypes<T>::Tensor::Index, 1> bcast; in operator ()() local
261 Eigen::array<typename TTypes<T>::Tensor::Index, 1> bcast; in operator ()() local
[all …]
Dops_util_test.cc124 static void VerifyBoundaries(bcast_struct bcast, error::Code code) { in VerifyBoundaries()
132 static void VerifyBcastValues(bcast_struct bcast) { in VerifyBcastValues()
178 bcast_struct bcast = {{2, 3, 1, 2, 0}, {0, 3}}; in TEST_F() local
184 bcast_struct bcast[] = { in TEST_F() local
196 bcast_struct bcast[] = { in TEST_F() local
208 bcast_struct bcast[] = { in TEST_F() local
220 bcast_struct bcast[] = { in TEST_F() local
231 bcast_struct bcast[] = { in TEST_F() local
242 bcast_struct bcast[] = { in TEST_F() local
252 bcast_struct bcast[] = { in TEST_F() local
[all …]
Dbcast_ops.cc46 BCast bcast(shapes[0], shapes[1]); in Compute() local
95 BCast bcast(shapes[0], shapes[1]); in Compute() local
Deigen_softmax.h51 Eigen::array<typename internal::traits<Input>::Index, 2> bcast; in eval() local
Deigen_softmax_test.cc40 Eigen::array<int, 2> bcast; in TEST() local
Dbroadcast_to_op.cc77 BCast bcast(BCast::FromShape(input_shape), BCast::FromShape(output_shape), in Compute() local
Dxent_op.cc49 BCast bcast(BCast::FromShape(logits_in.shape()), in Compute() local
Dcwise_ops_common.h64 BCast bcast; member
96 BCast* bcast = &state.bcast; in Compute() local
Dunravel_index_op.cc101 Eigen::array<Eigen::Index, 2> bcast({1, indices_tensor.NumElements()}); in Compute() local
Dpooling_ops_3d.cc258 Eigen::array<int, 5> bcast = {1, csize, rsize, psize, 1}; in launch() local
262 bcast; in launch() local
431 Eigen::array<int, 5> bcast = {1, csize, rsize, psize, 1}; in launch() local
435 bcast; in launch() local
Dsubstr_op.cc131 BCast bcast(BCast::FromShape(input_shape), BCast::FromShape(pos_shape)); in Compute() local
Dquantized_mul_op.cc292 BCast bcast(BCast::FromShape(x.shape()), BCast::FromShape(y.shape())); in Compute() local
Dquantized_add_op.cc465 BCast bcast(BCast::FromShape(x.shape()), BCast::FromShape(y.shape())); in Compute() local
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dbcast_ops.cc52 BCast bcast(shapes[0], shapes[1]); in Compile() local
102 BCast bcast(shapes[0], shapes[1]); in Compile() local
Dcwise_ops.cc43 BCast bcast(BCast::FromShape(lhs_shape), BCast::FromShape(rhs_shape), in Compile() local
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_rematerialization_test.cc65 auto bcast = builder.AddInstruction( in MakeRematerializableComputation() local
110 auto bcast = builder.AddInstruction( in MakeRematerializableWhileComputation() local
174 const HloInstruction* bcast = concat->operand(0); in TEST_F() local
442 auto bcast = builder.AddInstruction( in TEST_F() local
601 auto bcast = builder.AddInstruction( in TEST_P() local
Dbuffer_assignment_test.cc1765 auto bcast = builder.AddInstruction(HloInstruction::CreateBroadcast( in TEST_F() local
2022 const HloInstruction* bcast = m->entry_computation()->root_instruction(); in TEST_F() local
2089 const HloInstruction* bcast = m->entry_computation()->root_instruction(); in TEST_F() local
Dheap_simulator_test.cc176 HloInstruction* bcast = builder.AddInstruction( in TEST_F() local
/external/eigen/unsupported/test/
Dcxx11_tensor_forced_eval.cpp61 Eigen::array<int, 2> bcast; in test_const() local
/external/tensorflow/tensorflow/core/grappler/utils/
Dsymbolic_shapes.cc99 BCast bcast(ShapeDims(left), ShapeDims(right), in ShapesBroadcastable() local
115 BCast bcast(ShapeDims(left), ShapeDims(right), in ShapeAfterBroadcast() local
/external/wpa_supplicant_8/src/wps/
Dwps.c321 const u8 bcast[ETH_ALEN] = { 0xff, 0xff, 0xff, 0xff, 0xff, 0xff }; in wps_is_addr_authorized() local
/external/tensorflow/tensorflow/contrib/tensor_forest/kernels/
Dtree_utils.cc100 Eigen::array<Eigen::Index, 1> bcast; in GetTwoBestClassification() local
181 Eigen::array<Eigen::Index, 1> bcast; in GetTwoBestRegression() local
/external/libnl/lib/route/
Daddr.c977 int rtnl_addr_set_broadcast(struct rtnl_addr *addr, struct nl_addr *bcast) in rtnl_addr_set_broadcast()
/external/spirv-llvm/lib/SPIRV/libSPIRV/
DSPIRVEntry.h679 T* bcast(SPIRVEntry *E) { in bcast() function
/external/tensorflow/tensorflow/compiler/xla/tests/
Dconcat_test.cc596 auto bcast = Broadcast(y, {5}); in XLA_TEST_F() local

12