| /external/eigen/unsupported/Eigen/CXX11/src/Tensor/ |
| D | TensorBroadcasting.h | 23 template<typename Broadcast, typename XprType>
24 struct traits<TensorBroadcastingOp<Broadcast, XprType> > : public traits<XprType>
36 template<typename Broadcast, typename XprType>
37 struct eval<TensorBroadcastingOp<Broadcast, XprType>, Eigen::Dense>
39 typedef const TensorBroadcastingOp<Broadcast, XprType>& type;
42 template<typename Broadcast, typename XprType>
43 struct nested<TensorBroadcastingOp<Broadcast, XprType>, 1, typename eval<TensorBroadcastingOp<Broad…
45 typedef TensorBroadcastingOp<Broadcast, XprType> type;
67 template<typename Broadcast, typename XprType>
68 class TensorBroadcastingOp : public TensorBase<TensorBroadcastingOp<Broadcast, XprType>, ReadOnlyAc…
[all …]
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | const_op.cc | 54 0, xla::Broadcast(xla::ConstantR0<bool>(b, proto_.bool_val(0)), in Compile()
61 ctx->SetOutput(0, xla::Broadcast(xla::ConstantR0<float>( in Compile()
69 ctx->SetOutput(0, xla::Broadcast(xla::ConstantR0<double>( in Compile()
79 xla::Broadcast(xla::ConstantR0<xla::complex64>( in Compile()
90 xla::Broadcast(xla::ConstantR0<xla::complex128>( in Compile()
100 0, xla::Broadcast(xla::ConstantR0<int32>(b, proto_.int_val(0)), in Compile()
107 ctx->SetOutput(0, xla::Broadcast(xla::ConstantR0<int64>( in Compile()
|
| D | image_ops.cc | 49 auto zeros = xla::Broadcast(zero, shape.dim_sizes()); in RGBToHSV()
389 included_iou = xla::Select(xla::Broadcast(active_elem, {num_boxes}), in operator ()()
448 xla::Sort(/*keys=*/-xla::Broadcast(scores_input, {4}), in Compile()
491 y1 = xla::Broadcast(y1, {1}); in Compile()
492 y2 = xla::Broadcast(y2, {1}); in Compile()
493 x1 = xla::Broadcast(x1, {1}); in Compile()
494 x2 = xla::Broadcast(x2, {1}); in Compile()
495 area = xla::Broadcast(area, {1}); in Compile()
511 xla::Broadcast(xla::ConstantR0<bool>(builder, true), {num_boxes}); in Compile()
527 xla::Gt(scores, xla::Broadcast(score_thresh, {num_boxes})); in Compile()
[all …]
|
| D | matrix_set_diag_op.cc | 67 auto indicator = xla::Eq(iota_m, xla::Broadcast(iota_n, {m}), in Compile()
69 indicator = xla::Broadcast(indicator, batch_shape.dim_sizes()); in Compile()
78 diag = xla::Add(diag, xla::Broadcast(zero, input_shape.dim_sizes()), in Compile()
|
| D | binary_ops.cc | 72 std::tie(x, y) = XlaBinaryOp::Broadcast(x, y, broadcast_helper); in DivNoNanImpl()
90 std::tie(x, y) = XlaBinaryOp::Broadcast(x, y, broadcast_helper); in MulNoNanImpl()
113 std::tie(x, y) = XlaBinaryOp::Broadcast(x, y, broadcast_helper); in FloorDivImpl()
133 std::tie(x, y) = XlaBinaryOp::Broadcast(x, y, broadcast_helper); in XlogyImpl()
142 std::tie(x, y) = XlaBinaryOp::Broadcast(x, y, broadcast_helper); in XdivyImpl()
154 std::tie(x, y) = XlaBinaryOp::Broadcast(x, y, broadcast_helper); in FloorModImpl()
|
| D | relu_op.cc | 77 xla::Broadcast(XlaHelpers::Zero(b, input_type(0)), shape.dim_sizes()); in Compile()
93 xla::Broadcast(XlaHelpers::Zero(b, input_type(0)), shape.dim_sizes()); in Compile()
94 const auto six = xla::Broadcast( in Compile()
|
| D | clip_by_value_op.cc | 48 min = xla::Broadcast(min, shape.dim_sizes()); in Compile()
52 max = xla::Broadcast(max, shape.dim_sizes()); in Compile()
|
| D | sparse_to_dense_op.cc | 71 sparse_values = Broadcast(sparse_values, {num_elems}); in Compile()
74 auto buffer = Broadcast(default_value, output_shape.dim_sizes()); in Compile()
|
| D | matrix_band_part_op.cc | 82 indicator = xla::Broadcast(indicator, batch_shape.dim_sizes()); in Compile()
86 indicator, input, xla::Broadcast(zero_input, input_shape.dim_sizes())); in Compile()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | convolution_group_converter_test.cc | 61 op::Select(op::Eq(op::Broadcast(op::Constant()), in TEST_F()
62 op::Broadcast(op::Constant())), in TEST_F()
63 op::Broadcast(op::Reshape(op::Parameter())), in TEST_F()
64 op::Broadcast(op::Constant()))); in TEST_F()
|
| D | hlo_rematerialization_test.cc | 172 ASSERT_THAT(slice, op::Slice(op::Concatenate(op::Broadcast(_), _))); in TEST_F()
188 EXPECT_THAT(remat_bcast, op::Broadcast(::testing::Ne(bcast))); in TEST_F()
495 EXPECT_THAT(add_2->operand(0), op::Broadcast(param)); in TEST_F()
497 EXPECT_THAT(add_3->operand(0), op::Broadcast(param)); in TEST_F()
499 EXPECT_THAT(add_4->operand(0), op::Broadcast(param)); in TEST_F()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | cudnn_fused_conv_rewriter.cc | 46 using match::Broadcast; in FindConvWithRelu()
62 auto zero_pattern = Broadcast(match::ConstantScalar(0)); in FindConvWithRelu()
79 const auto bias_pattern = Broadcast(&bias_broadcast_instr, Op(&bias)); in FindConvWithRelu()
81 auto alpha_pattern = Broadcast(Constant(&alpha_conv_instr)); in FindConvWithRelu()
88 auto alpha_pattern = Broadcast(Constant(&alpha_side_input_instr)); in FindConvWithRelu()
|
| D | instruction_fusion_test.cc | 172 op::Reduce(op::Broadcast(op::Constant()), op::Constant())); in TEST_F()
331 op::Broadcast(op::Constant()))); in TEST_F()
465 op::Broadcast(op::Constant()))); in TEST_F()
537 op::Add(op::Broadcast(op::Add(op::Parameter(), op::Constant())), in TEST_F()
637 op::Reduce(op::Broadcast(op::Parameter()), op::Constant())); in TEST_F()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | xla_resource.cc | 132 xla::Broadcast(XlaHelpers::Zero(builder, type_), shape_.dim_sizes()); in SetZeroValue()
139 value_ = xla::Broadcast(XlaHelpers::Zero(builder, type_), in SetZeroValue()
148 xla::Tuple(builder, {xla::Broadcast(XlaHelpers::Zero(builder, type_), in SetZeroValue()
173 xla::Broadcast(XlaHelpers::Zero(builder, type_), ta_shape.dim_sizes()); in GetOrCreateTensorArrayGradient()
|
| D | xla_helpers.cc | 103 xla::Broadcast(on_value, output_shape.dim_sizes()), in OneHot()
104 xla::Broadcast(off_value, output_shape.dim_sizes())); in OneHot()
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | broadcast_simple_test.cc | 110 Broadcast(ConstantR0<float>(&b, 1.5), {}); in XLA_TEST_F()
116 Broadcast(ConstantR0<float>(&b, 2.25), {2, 3}); in XLA_TEST_F()
128 Broadcast(src, {2, 3}); in XLA_TEST_F()
136 Broadcast(ConstantR0<float>(&b, 2.25), {2, 0}); in XLA_TEST_F()
143 Broadcast(ConstantR0<float>(&b, 2.25), {0, 2}); in XLA_TEST_F()
150 Broadcast(ConstantR1<float>(&b, {1, 2, 3}), {2});
268 Broadcast(ConstantR1<float>(&b, {}), {2}); in XLA_TEST_F()
276 Broadcast(ConstantR1<float>(&b, {1, 2, 3}), {0});
715 auto r3 = Broadcast(r0, {2, 2, 2}); in XLA_TEST_F()
|
| D | outfeed_in_nested_computation_test.cc | 37 XlaOp some_buffer = Broadcast(ConstantR0<int32_t>(&b, 0), {10, 5}); in XLA_TEST_F()
59 XlaOp sum = Add(some_input, Broadcast(loop_counter, {2})); in XLA_TEST_F()
|
| /external/libchrome/base/synchronization/ |
| D | condition_variable_unittest.cc | 319 queue.work_is_available()->Broadcast(); // Make them all try. in TEST_F()
340 queue.work_is_available()->Broadcast(); // Signal all threads. in TEST_F()
380 queue.work_is_available()->Broadcast(); in TEST_F()
395 queue.work_is_available()->Broadcast(); // Force check for shutdown. in TEST_F()
440 queue.work_is_available()->Broadcast(); // Start up all threads. in TEST_F()
491 queue.work_is_available()->Broadcast(); // Force check for shutdown. in TEST_F()
534 work_is_available_.Broadcast(); // Tell them all to terminate. in ~WorkQueue()
|
| D | condition_variable.h | 101 void Broadcast();
|
| /external/syzkaller/pkg/ipc/ |
| D | gate.go | 70 g.cv.Broadcast()
73 g.cv.Broadcast()
|
| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| D | qr.cc | 105 *tau = Select(sigma_is_zero, Broadcast(zero, batch_dims), in House()
108 Select(sigma_is_zero, Broadcast(one, batch_dims), alpha - *beta); in House()
110 auto e_k = Broadcast(ConvertElementType(Eq(iota, k), type), in House()
205 auto mask = Broadcast(ConvertElementType(Eq(iota, j), type), in QRBlock()
350 auto q = Broadcast(IdentityMatrix(builder, type, m, m), batch_dims); in QRDecomposition()
|
| D | matrix.cc | 45 auto indicator = Eq(a, Broadcast(b, {m}), /*broadcast_dimensions=*/{0}); in IdentityMatrix()
64 auto indicator = Eq(b, Broadcast(a, {m}), /*broadcast_dimensions=*/{0}); in GetMatrixDiagonal()
65 auto mask = Broadcast(indicator, major_dims); in GetMatrixDiagonal()
104 indicator = Ge(b, Broadcast(a, {m}), /*broadcast_dimensions=*/{0}); in TriangleMask()
105 return Broadcast(indicator, major_dims); in TriangleMask()
|
| D | quantize.h | 104 auto broadcast_input = Broadcast(input, {unpack_size});
132 broadcast_input, Transpose(Broadcast(shift_bits, shape.dimensions()),
|
| /external/libchrome/base/test/ |
| D | scoped_task_environment.cc | 298 can_run_tasks_cv_.Broadcast(); in AllowRunTasks()
341 task_completed_.Broadcast(); in RunOrSkipTask()
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder_test.cc | 186 EXPECT_THAT(root, op::Add(op::Parameter(), op::Broadcast(op::Constant()))); in TEST_F()
202 EXPECT_THAT(root, op::Add(op::Parameter(0), op::Broadcast(op::Parameter(1)))); in TEST_F()
274 op::Broadcast(op::Reshape(op::Parameter(1))))); in TEST_F()
298 EXPECT_THAT(root, op::Add(op::Broadcast(op::Parameter(0)), in TEST_F()
299 op::Broadcast(op::Reshape(op::Parameter(1))))); in TEST_F()
309 EXPECT_THAT(root, op::Broadcast()); in TEST_F()
319 op::Broadcast(op::Reshape(op::Broadcast()))); in TEST_F()
|