| /external/tensorflow/tensorflow/core/util/ |
| D | matmul_bcast_test.cc | 38 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
118 MatMulBCast bcast({3, 1, 5, 3}, {1, 4, 3, 7}); in TEST() local
130 MatMulBCast bcast({3, 1, 5, 3}, {2, 1, 2, 3, 7}); in TEST() local
|
| D | bcast.h | 36 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/ |
| D | ops_util_test.cc | 126 static void VerifyBoundaries(bcast_struct bcast, error::Code code) { in VerifyBoundaries()
134 static void VerifyBcastValues(bcast_struct bcast) { in VerifyBcastValues()
180 bcast_struct bcast = {{2, 3, 1, 2, 0}, {0, 3}}; in TEST_F() local
186 bcast_struct bcast[] = { in TEST_F() local
198 bcast_struct bcast[] = { in TEST_F() local
210 bcast_struct bcast[] = { in TEST_F() local
222 bcast_struct bcast[] = { in TEST_F() local
233 bcast_struct bcast[] = { in TEST_F() local
244 bcast_struct bcast[] = { in TEST_F() local
254 bcast_struct bcast[] = { in TEST_F() local
[all …]
|
| D | bcast_ops.cc | 46 BCast bcast(shapes[0], shapes[1]); in Compute() local
95 BCast bcast(shapes[0], shapes[1]); in Compute() local
|
| D | training_ops_gpu.cu.cc | 341 Eigen::array<typename TTypes<T>::Tensor::Index, 1> bcast; in operator ()() local
630 Eigen::array<typename TTypes<T>::Tensor::Index, 1> bcast; in operator ()() local
665 Eigen::array<typename TTypes<T>::Tensor::Index, 1> bcast; in operator ()() local
701 Eigen::array<typename TTypes<T>::Tensor::Index, 1> bcast; in operator ()() local
741 Eigen::array<typename TTypes<T>::Tensor::Index, 1> bcast; in operator ()() local
815 Eigen::array<typename TTypes<T>::Tensor::Index, 1> bcast; in operator ()() local
836 Eigen::array<typename TTypes<T>::Tensor::Index, 1> bcast; in operator ()() local
912 Eigen::array<typename TTypes<T>::Tensor::Index, 1> bcast; in operator ()() local
943 Eigen::array<typename TTypes<T>::Tensor::Index, 1> bcast; in operator ()() local
1028 Eigen::array<typename TTypes<T>::Tensor::Index, 1> bcast; in operator ()() local
[all …]
|
| D | broadcast_to_op.cc | 79 BCast bcast(BCast::FromShape(input_shape), BCast::FromShape(output_shape), in Compute() local
|
| D | xent_op.cc | 46 BCast bcast(BCast::FromShape(logits_in.shape()), in Compute() local
|
| D | cwise_ops_common.h | 60 BCast bcast; member
145 auto& bcast = state.bcast; in Compute() local
|
| D | pooling_ops_3d.cc | 254 Eigen::array<int, 5> bcast = {1, csize, rsize, psize, 1}; in launch() local
258 bcast; in launch() local
427 Eigen::array<int, 5> bcast = {1, csize, rsize, psize, 1}; in launch() local
431 bcast; in launch() local
|
| D | unravel_index_op.cc | 112 Eigen::array<Eigen::Index, 2> bcast( in Compute() local
|
| D | random_binomial_op.cc | 342 tensorflow::BCast bcast(counts_tensor.shape().dim_sizes(), in Compute() local
459 tensorflow::BCast bcast(counts_tensor.shape().dim_sizes(), in Compute() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_rematerialization_test_utils.h | 64 auto bcast = builder.AddInstruction( variable
109 auto bcast = builder.AddInstruction( variable
|
| D | hlo_rematerialization_test.cc | 74 const HloInstruction* bcast = concat->operand(0); in TEST_F() local
364 auto bcast = builder.AddInstruction( in TEST_F() local
521 auto bcast = builder.AddInstruction( in TEST_P() local
779 const HloInstruction* bcast = concat->operand(0)->operand(0); in TEST_F() local
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | bcast_ops.cc | 52 BCast bcast(shapes[0], shapes[1]); in Compile() local
102 BCast bcast(shapes[0], shapes[1]); in Compile() local
|
| D | cwise_ops.cc | 43 BCast bcast(BCast::FromShape(lhs_shape), BCast::FromShape(rhs_shape), in Compile() local
|
| D | matrix_triangular_solve_op.cc | 45 MatMulBCast bcast(BCast::FromShape(lhs_shape), BCast::FromShape(rhs_shape)); in Compile() local
|
| D | select_op.cc | 104 BCast bcast(bcast_then_else.output_shape(), BCast::FromShape(cond_shape), in Compile() local
|
| /external/tensorflow/tensorflow/compiler/tf2xla/lib/ |
| D | broadcast.cc | 102 BCast bcast(BCast::FromShape(lhs_tf_shape), BCast::FromShape(rhs_tf_shape)); in BroadcastOpsToSame() local
|
| /external/eigen/unsupported/test/ |
| D | cxx11_tensor_forced_eval.cpp | 61 Eigen::array<int, 2> bcast; in test_const() local
|
| /external/tensorflow/tensorflow/core/grappler/utils/ |
| D | symbolic_shapes.cc | 102 BCast bcast(ShapeDims(left), ShapeDims(right), in ShapesBroadcastable() local
118 BCast bcast(ShapeDims(left), ShapeDims(right), in ShapeAfterBroadcast() local
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | unroll_batch_matmul.cc | 157 const tensorflow::MatMulBCast& bcast, int rows, int cols, Type element_type, in createMatMulOps()
274 tensorflow::MatMulBCast bcast(absl::InlinedVector<tensorflow::int64, 4>( in matchAndRewrite() local
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | mkl_remapper_test.cc | 184 #define CREATE_CONV2DFUSION_TEST(data_format, addop, activation, bcast) \ argument
191 #define CREATE_CONV2DFUSION_ADD_ACTIVATION_TEST(data_format, addop, bcast) \ argument
|
| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_batch_matmul_op.cc | 97 MatMulBCast bcast(lhs.shape().dim_sizes(), rhs.shape().dim_sizes()); in Compute() local
|
| /external/tensorflow/tensorflow/core/kernels/linalg/ |
| D | banded_triangular_solve_op.cc | 221 MatMulBCast bcast(in0.shape().dim_sizes(), in1.shape().dim_sizes()); in Compute() local
|
| /external/wpa_supplicant_8/src/wps/ |
| D | wps.c | 321 const u8 bcast[ETH_ALEN] = { 0xff, 0xff, 0xff, 0xff, 0xff, 0xff }; in wps_is_addr_authorized() local
|