Home
last modified time | relevance | path

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

12

/external/tensorflow/tensorflow/core/util/
Dmatmul_bcast_test.cc38 MatMulBCast bcast({1, 5, 3}, {4, 3, 7}); in TEST() local
51 MatMulBCast bcast({5, 3}, {3, 7}); in TEST() local
64 MatMulBCast bcast({2, 4, 6, 5, 3}, {2, 4, 6, 3, 7}); in TEST() local
114 MatMulBCast bcast({3, 1, 5, 3}, {1, 4, 3, 7}); in TEST() local
126 MatMulBCast bcast({3, 1, 5, 3}, {2, 1, 2, 3, 7}); in TEST() local
Dbcast.h36 const gtl::InlinedVector<int64, 4>& bcast, in ComputeBatchIndices()
95 const Vec& bcast(int i) const { return bcast_[i]; } in bcast() function
/external/tensorflow/tensorflow/core/kernels/
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 …]
Dtraining_ops_gpu.cu.cc114 Eigen::array<typename TTypes<T>::Tensor::Index, 1> bcast; in operator ()() local
130 Eigen::array<typename TTypes<T>::Tensor::Index, 1> bcast; in operator ()() local
144 Eigen::array<typename TTypes<T>::Tensor::Index, 1> bcast; in operator ()() local
165 Eigen::array<typename TTypes<T>::Tensor::Index, 1> bcast; in operator ()() local
193 Eigen::array<typename TTypes<T>::Tensor::Index, 1> bcast; in operator ()() local
229 Eigen::array<typename TTypes<T>::Tensor::Index, 1> bcast; in operator ()() local
265 Eigen::array<typename TTypes<T>::Tensor::Index, 1> bcast; in operator ()() local
286 Eigen::array<typename TTypes<T>::Tensor::Index, 1> bcast; in operator ()() local
360 Eigen::array<typename TTypes<T>::Tensor::Index, 1> bcast; in operator ()() local
391 Eigen::array<typename TTypes<T>::Tensor::Index, 1> bcast; in operator ()() local
[all …]
Dbcast_ops.cc46 BCast bcast(shapes[0], shapes[1]); in Compute() local
95 BCast bcast(shapes[0], shapes[1]); in Compute() local
Dbroadcast_to_op.cc79 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
Dunravel_index_op.cc101 Eigen::array<Eigen::Index, 2> bcast({1, indices_tensor.NumElements()}); in Compute() local
Dcwise_ops_common.h66 BCast bcast; member
151 auto& bcast = state.bcast; 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
Drandom_binomial_op.cc340 tensorflow::BCast bcast(counts_tensor.shape().dim_sizes(), in Compute() local
Dcwise_op_select.cc172 BCast bcast( in Compute() local
Dmkl_batch_matmul_op.cc99 MatMulBCast bcast(lhs.shape().dim_sizes(), rhs.shape().dim_sizes()); in Compute() local
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_rematerialization_test_utils.h64 auto bcast = builder.AddInstruction( variable
109 auto bcast = builder.AddInstruction( variable
Dhlo_rematerialization_test.cc70 const HloInstruction* bcast = concat->operand(0); in TEST_F() local
340 auto bcast = builder.AddInstruction( in TEST_F() local
499 auto bcast = builder.AddInstruction( in TEST_P() 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
Dmatrix_triangular_solve_op.cc45 MatMulBCast bcast(BCast::FromShape(lhs_shape), BCast::FromShape(rhs_shape)); in Compile() local
Dselect_op.cc104 BCast bcast(bcast_then_else.output_shape(), BCast::FromShape(cond_shape), in Compile() local
/external/tensorflow/tensorflow/compiler/tf2xla/lib/
Dbroadcast.cc102 BCast bcast(BCast::FromShape(lhs_tf_shape), BCast::FromShape(rhs_tf_shape)); in BroadcastOpsToSame() 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.cc102 BCast bcast(ShapeDims(left), ShapeDims(right), in ShapesBroadcastable() local
118 BCast bcast(ShapeDims(left), ShapeDims(right), in ShapeAfterBroadcast() local
/external/tensorflow/tensorflow/compiler/mlir/lite/transforms/
Dunroll_batch_matmul.cc162 const tensorflow::MatMulBCast& bcast, int rows, int cols, Type element_type, in createMatMulOps()
279 tensorflow::MatMulBCast bcast(absl::InlinedVector<tensorflow::int64, 4>( in matchAndRewrite() 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

12