Home
last modified time | relevance | path

Searched refs:CreateScalarAddComputation (Results 1 – 25 of 25) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/tests/
Dvector_ops_reduce_test.cc57 auto sum_reducer = CreateScalarAddComputation(F32, &builder_); in TEST_F()
68 auto sum_reducer = CreateScalarAddComputation(F32, &builder_); in TEST_F()
104 auto sum_reducer = CreateScalarAddComputation(F32, &builder_); in TEST_F()
120 auto sum_reducer = CreateScalarAddComputation(F32, &builder_); in TEST_F()
134 auto sum_reducer = CreateScalarAddComputation(F32, &builder_); in TEST_F()
145 auto sum_reducer = CreateScalarAddComputation(F32, &builder_); in TEST_F()
157 auto sum_reducer = CreateScalarAddComputation(F32, &builder_); in TEST_F()
168 auto sum_reducer = CreateScalarAddComputation(F32, &builder_); in TEST_F()
177 auto sum_reducer = CreateScalarAddComputation(F32, &builder_); in XLA_TEST_F()
186 auto sum_reducer = CreateScalarAddComputation(F32, &builder_); in TEST_F()
[all …]
Dreduce_test.cc94 XlaComputation add_f32 = CreateScalarAddComputation(F32, &builder); in RunR1ToR0Test()
204 XlaComputation add_f32 = CreateScalarAddComputation(F32, &builder); in RunR2ToR0Test()
231 XlaComputation add_f32 = CreateScalarAddComputation(F32, &builder); in RunR2ToR1Test()
441 XlaComputation add_f32 = CreateScalarAddComputation(F32, &builder); in XLA_TEST_F()
471 XlaComputation add_f32 = CreateScalarAddComputation(F32, &builder); in XLA_TEST_F()
502 XlaComputation add_f32 = CreateScalarAddComputation(F32, &builder); in XLA_TEST_F()
518 XlaComputation add_f32 = CreateScalarAddComputation(F32, &builder); in XLA_TEST_F()
563 auto add = CreateScalarAddComputation(F32, &builder); in XLA_TEST_F()
643 auto add = CreateScalarAddComputation(F32, &builder); in XLA_TEST_F()
654 auto add = CreateScalarAddComputation(F32, &builder); in XLA_TEST_F()
[all …]
Dreduce_window_test.cc79 CreateScalarAddComputation(FloatType(), &builder_), in ReduceWindowAdd()
115 CreateScalarAddComputation(FloatType(), &builder_), in TEST_P()
130 ReduceWindow(input, init, CreateScalarAddComputation(FloatType(), &builder_), in TEST_P()
636 ? CreateScalarAddComputation(FloatType(), &b) in DoIt()
1037 ? CreateScalarAddComputation(FloatType(), &b) in TEST_P()
1174 ? CreateScalarAddComputation(FloatType(), &b) in DoIt()
1367 ? CreateScalarAddComputation(FloatType(), &b) in TEST_P()
Dmap_test.cc356 Map(&builder, {param0, param1}, CreateScalarAddComputation(F32, &builder), in TEST_F()
380 Map(&builder, {param0, param1}, CreateScalarAddComputation(S32, &builder), in XLA_TEST_F()
406 Map(&builder, {param0, param1}, CreateScalarAddComputation(S32, &builder), in XLA_TEST_F()
Dselect_and_scatter_test.cc56 add_s32_ = CreateScalarAddComputation(S32, &builder_); in SelectAndScatterTest()
58 add_f32_ = CreateScalarAddComputation(F32, &builder_); in SelectAndScatterTest()
Dcholesky_test.cc189 CreateScalarAddComputation(F32, &builder), {0, 1, 2}); in XLA_TEST_P()
Dbatch_normalization_test.cc139 XlaComputation add = CreateScalarAddComputation(F32, &builder); in XLA_TEST_P()
153 XlaComputation add = CreateScalarAddComputation(F32, &builder); in XLA_TEST_P()
179 XlaComputation add = CreateScalarAddComputation(F32, &builder); in XLA_TEST_P()
Dvector_ops_simple_test.cc152 auto add = CreateScalarAddComputation(F32, &builder); in XLA_TEST_F()
Dpad_test.cc452 XlaComputation add = CreateScalarAddComputation(FloatType(), &b); in XLA_TEST_P()
Dwhile_test.cc144 CreateScalarAddComputation(S32, &builder), {0}); in XLA_TEST_F()
/external/tensorflow/tensorflow/compiler/xla/client/lib/
Dsvd.cc139 CreateScalarAddComputation(x_shape.element_type(), builder), in HouseRow()
205 CreateScalarAddComputation(x_shape.element_type(), builder), in HouseCol()
524 CreateScalarAddComputation(d_shape.element_type(), builder), in OneSidedJacobiUpdate()
534 CreateScalarAddComputation(d_shape.element_type(), builder), in OneSidedJacobiUpdate()
550 CreateScalarAddComputation(d_shape.element_type(), builder), in OneSidedJacobiUpdate()
560 CreateScalarAddComputation(d_shape.element_type(), builder), in OneSidedJacobiUpdate()
580 CreateScalarAddComputation(shape.element_type(), builder), in ComputeFrobeniusNorms()
585 CreateScalarAddComputation(shape.element_type(), builder), in ComputeFrobeniusNorms()
776 CreateScalarAddComputation(shape.element_type(), builder), in SortBySingularValuesAndPostProcessing()
788 CreateScalarAddComputation(shape.element_type(), builder), in SortBySingularValuesAndPostProcessing()
Dself_adjoint_eig.cc176 CreateScalarAddComputation(w_shape.element_type(), builder), in Update()
183 CreateScalarAddComputation(w_shape.element_type(), builder), in Update()
201 CreateScalarAddComputation(shape.element_type(), builder), in ComputeFrobeniusNorms()
206 CreateScalarAddComputation(shape.element_type(), builder), in ComputeFrobeniusNorms()
Dpooling.cc66 ReduceWindow(padded_ones, zero, CreateScalarAddComputation(dtype, b), in AvgPoolDivideByCountWithGeneralPadding()
82 auto add_computation = CreateScalarAddComputation(accumulation_type, b); in ComputeSums()
274 CreateScalarAddComputation(dtype, b), kernel_size, in AvgPoolGrad()
Darithmetic.h28 XlaComputation CreateScalarAddComputation(PrimitiveType type,
Darithmetic.cc54 XlaComputation CreateScalarAddComputation(PrimitiveType type, in CreateScalarAddComputation() function
Dsvd_test.cc121 CreateScalarAddComputation(F32, builder)) / in GetAverageAbsoluteError()
Dself_adjoint_eig_test.cc122 CreateScalarAddComputation(F32, builder)) / in GetAverageAbsoluteError()
Dqr.cc98 CreateScalarAddComputation(type, builder), {minor_dim}); in House()
Dmatrix.cc72 : CreateScalarAddComputation(shape.element_type(), builder); in GetMatrixDiagonal()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/tests/
Dcpu_infeed_test.cc140 CreateScalarAddComputation(F32, &builder), {0}); in TEST_F()
230 CreateScalarAddComputation(F32, &builder), {0}); in TEST_F()
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dbucketize_op.cc56 /*computation=*/xla::CreateScalarAddComputation(xla::S32, builder), in Compile()
Dresampler_ops.cc188 xla::CreateScalarAddComputation(xla_type, ctx->builder()), in ScatterToGradData()
464 xla::CreateScalarAddComputation(data_type, ctx->builder()), in CalculateGradWarp()
475 xla::CreateScalarAddComputation(data_type, ctx->builder()), in CalculateGradWarp()
Dconv_op_helpers.cc160 CreateScalarAddComputation(elem_type, builder), in ContractFilterForDepthwiseBackprop()
Dpooling_ops.cc336 auto scatter = CreateScalarAddComputation(element_type, ctx->builder()); in Compile()
Dimage_ops.cc552 /*computation=*/CreateScalarAddComputation(xla::S32, builder), in Compile()