| /aosp_15_r20/external/tensorflow/tensorflow/core/util/ |
| H A D | strided_slice_op_test.cc | 98 StridedSliceAssignBCast bcast(test_pair.from, test_pair.to); in TEST() local
118 StridedSliceAssignBCast bcast(test_pair.from, test_pair.to); in TEST() local
137 StridedSliceAssignBCast bcast(test_pair.from, test_pair.to); in TEST() local
160 StridedSliceAssignBCast bcast(test_pair.from, test_pair.to); in TEST() local
198 StridedSliceAssignBCast bcast(test_pair.from, test_pair.to); in TEST() local
215 StridedSliceAssignBCast bcast(test_pair.from, test_pair.to); in TEST() local
240 StridedSliceAssignBCast bcast(test_pair.from, test_pair.to); in TEST() local
264 StridedSliceAssignBCast bcast(test_pair.from, test_pair.to); in TEST() local
292 StridedSliceAssignBCast bcast(test_pair.from, test_pair.to); in TEST() local
315 StridedSliceAssignBCast bcast(test_pair.from, test_pair.to); in TEST() local
[all …]
|
| H A 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
|
| H A D | bcast.h | 36 const gtl::InlinedVector<int64_t, 4>& bcast, in ComputeBatchIndices()
95 const Vec& bcast(int i) const { return bcast_[i]; } in bcast() function
|
| H A D | strided_slice_op.h | 135 const Vec& bcast() const { return bcast_; } in bcast() function
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A 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 …]
|
| H A D | training_ops_gpu.cu.cc | 266 Eigen::array<typename TTypes<T>::Tensor::Index, 1> bcast; in operator ()() local
557 Eigen::array<typename TTypes<T>::Tensor::Index, 1> bcast; in operator ()() local
592 Eigen::array<typename TTypes<T>::Tensor::Index, 1> bcast; in operator ()() local
628 Eigen::array<typename TTypes<T>::Tensor::Index, 1> bcast; in operator ()() local
668 Eigen::array<typename TTypes<T>::Tensor::Index, 1> bcast; in operator ()() local
742 Eigen::array<typename TTypes<T>::Tensor::Index, 1> bcast; in operator ()() local
763 Eigen::array<typename TTypes<T>::Tensor::Index, 1> bcast; in operator ()() local
905 Eigen::array<typename TTypes<T>::Tensor::Index, 1> bcast; in operator ()() local
936 Eigen::array<typename TTypes<T>::Tensor::Index, 1> bcast; in operator ()() local
1021 Eigen::array<typename TTypes<T>::Tensor::Index, 1> bcast; in operator ()() local
[all …]
|
| H A D | bcast_ops.cc | 46 BCast bcast(shapes[0], shapes[1]); in Compute() local
95 BCast bcast(shapes[0], shapes[1]); in Compute() local
|
| H A D | cwise_ops_common.h | 60 BCast bcast; member
155 auto& bcast = state.bcast; in Compute() local
|
| H A 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
|
| H A D | pooling_ops_3d.cc | 260 bcast; in launch() local
455 bcast; in launch() local
|
| H A D | xent_op.cc | 48 BCast bcast(BCast::FromShape(logits_in.shape()), in Compute() local
|
| H A D | broadcast_to_op.cc | 76 BCast bcast(BCast::FromShape(input_shape), BCast::FromShape(output_shape), in Compute() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/ |
| H A D | hlo_rematerialization_test_utils.h | 64 auto bcast = builder.AddInstruction( variable
109 auto bcast = builder.AddInstruction( variable
|
| H A D | all_gather_broadcast_reorder.cc | 52 HloBroadcastInstruction *bcast = in Run() local
190 HloInstruction *bcast = computation->AddInstruction( in Run() local
|
| H A 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
821 const HloInstruction* bcast = concat->operand(0)->operand(0); in TEST_F() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| H A D | bcast_ops.cc | 53 BCast bcast(shapes[0], shapes[1]); in Compile() local
110 BCast bcast(shapes[0], shapes[1]); in Compile() local
|
| H A D | matrix_triangular_solve_op.cc | 45 MatMulBCast bcast(BCast::FromShape(lhs_shape), BCast::FromShape(rhs_shape)); in Compile() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tfrt/jit/transforms/ |
| H A D | tf_jitrt_math_approximation.cc | 150 auto bcast = [&](Value value) -> Value { in ldexp() local
201 auto bcast = [&](Value value) -> Value { in matchAndRewrite() local
263 auto bcast = [&](Value value) -> Value { in matchAndRewrite() local
|
| /aosp_15_r20/packages/apps/Protips/src/com/android/protips/ |
| D | ProtipWidget.java | 148 final Intent bcast = new Intent(context, ProtipWidget.class); in onReceiveAsync() local
196 Intent bcast = new Intent(context, ProtipWidget.class); in buildUpdate() local
|
| /aosp_15_r20/tools/test/connectivity/acts_tests/tests/google/net/ |
| D | DhcpServerTest.py | 736 def _run_win10_request_renewing(self, bcast): argument
810 def _run_debian_renewing(self, bcast): argument
833 def _assert_standard_offer_or_ack(self, resp, renewing=False, bcast=False, argument
877 def _request_address(self, hwaddr, bcast=True): argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/grappler/utils/ |
| H A D | symbolic_shapes.cc | 102 BCast bcast(ShapeDims(left), ShapeDims(right), in ShapesBroadcastable() local
118 BCast bcast(ShapeDims(left), ShapeDims(right), in ShapeAfterBroadcast() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/lib/Analysis/ |
| H A D | shape_component_analysis.cc | 88 } else if (auto bcast = in visit() local
126 } else if (auto bcast = value.getDefiningOp<shape::BroadcastOp>()) { in visit() local
201 } else if (auto bcast = value.getDefiningOp<shape::BroadcastOp>()) { in visit() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/lib/ |
| H A D | broadcast.cc | 45 tensorflow::BCast bcast(tensorflow::BCast::FromShape(lhs_tf_shape), in BroadcastOpsToSame() local
|
| /aosp_15_r20/external/eigen/unsupported/test/ |
| H A D | cxx11_tensor_forced_eval.cpp | 61 Eigen::array<int, 2> bcast; in test_const() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| H A D | gemm_broadcast_folding_rewriter.cc | 42 HloInstruction *bcast; in HandleCustomCall() local
|