Home
last modified time | relevance | path

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

12

/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.cc97 XlaComputation add_f32 = CreateScalarAddComputation(F32, &builder); in RunR1ToR0Test()
205 XlaComputation add_f32 = CreateScalarAddComputation(F32, &builder); in RunR2ToR0Test()
232 XlaComputation add_f32 = CreateScalarAddComputation(F32, &builder); in RunR2ToR1Test()
442 XlaComputation add_f32 = CreateScalarAddComputation(F32, &builder); in XLA_TEST_F()
472 XlaComputation add_f32 = CreateScalarAddComputation(F32, &builder); in XLA_TEST_F()
503 XlaComputation add_f32 = CreateScalarAddComputation(F32, &builder); in XLA_TEST_F()
519 XlaComputation add_f32 = CreateScalarAddComputation(F32, &builder); in XLA_TEST_F()
564 auto add = CreateScalarAddComputation(F32, &builder); in XLA_TEST_F()
644 auto add = CreateScalarAddComputation(F32, &builder); in XLA_TEST_F()
655 auto add = CreateScalarAddComputation(F32, &builder); in XLA_TEST_F()
[all …]
Dcholesky_test.cc254 CreateScalarAddComputation(F32, &builder), {0, 1, 2}); in XLA_TEST_P()
299 CreateScalarAddComputation(F32, &builder), {0, 1, 2}); in XLA_TEST_P()
Dreduce_window_test.cc79 CreateScalarAddComputation(FloatType(), &builder_), in ReduceWindowAdd()
115 CreateScalarAddComputation(FloatType(), &builder_), in XLA_TEST_P()
130 ReduceWindow(input, init, CreateScalarAddComputation(FloatType(), &builder_), in XLA_TEST_P()
652 ? CreateScalarAddComputation(FloatType(), &b) in DoIt()
1053 ? CreateScalarAddComputation(FloatType(), &b) in XLA_TEST_P()
1260 ? CreateScalarAddComputation(FloatType(), &b) in DoIt()
1461 ? CreateScalarAddComputation(FloatType(), &b) in XLA_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()
Dbatch_normalization_test.cc140 XlaComputation add = CreateScalarAddComputation(F32, &builder); in XLA_TEST_P()
154 XlaComputation add = CreateScalarAddComputation(F32, &builder); in XLA_TEST_P()
180 XlaComputation add = CreateScalarAddComputation(F32, &builder); in XLA_TEST_P()
Dvector_ops_simple_test.cc152 auto add = CreateScalarAddComputation(F32, &builder); in XLA_TEST_F()
Ddynamism_inference_test.cc173 XlaComputation add_s32 = CreateScalarAddComputation(S32, &b); in TEST_F()
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Din_topk_op.cc81 xla::CreateScalarAddComputation(xla::F32, xla_builder), {1}); in Compile()
94 xla::CreateScalarAddComputation(xla::S32, xla_builder), {1}); in Compile()
Dbucketize_op.cc56 /*computation=*/xla::CreateScalarAddComputation(xla::S32, builder), in Compile()
Dwhere_op.cc57 xla::CreateScalarAddComputation(xla::S32, ctx->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()
Dunique_op.cc210 xla::CreateScalarAddComputation(xla::S32, ctx->builder())); in DataOutputFastPath()
/external/tensorflow/tensorflow/compiler/xla/client/lib/
Dsvd.cc134 CreateScalarAddComputation(x_shape.element_type(), builder), in HouseRow()
199 CreateScalarAddComputation(x_shape.element_type(), builder), in HouseCol()
515 CreateScalarAddComputation(d_shape.element_type(), builder), in OneSidedJacobiUpdate()
525 CreateScalarAddComputation(d_shape.element_type(), builder), in OneSidedJacobiUpdate()
541 CreateScalarAddComputation(d_shape.element_type(), builder), in OneSidedJacobiUpdate()
551 CreateScalarAddComputation(d_shape.element_type(), builder), in OneSidedJacobiUpdate()
763 CreateScalarAddComputation(shape.element_type(), builder), in SortBySingularValuesAndPostProcessing()
775 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()
277 CreateScalarAddComputation(dtype, b), kernel_size, in AvgPoolGrad()
Darithmetic.h35 XlaComputation CreateScalarAddComputation(PrimitiveType type,
Darithmetic.cc49 XlaComputation CreateScalarAddComputation(PrimitiveType type, in CreateScalarAddComputation() function
96 : CreateScalarAddComputation(type, builder); in CreateScalarIdentityWithZeroComputation()
Dsvd_test.cc106 CreateScalarAddComputation(F32, builder)) / in GetAverageAbsoluteError()
Dself_adjoint_eig_test.cc122 CreateScalarAddComputation(F32, builder)) / in GetAverageAbsoluteError()
Dmatrix.cc501 CreateScalarAddComputation(x_shape.element_type(), builder), in Einsum()
507 CreateScalarAddComputation(y_shape.element_type(), builder), in Einsum()
/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/xla/service/
Dqr_expander.cc108 CreateScalarAddComputation( in House()
126 CreateScalarAddComputation(type, builder), {minor_dim}); in House()
/external/tensorflow/tensorflow/core/tpu/kernels/
Dtopk_ops.cc81 xla::XlaComputation add = CreateScalarAddComputation(xla::S32, builder); in CreateKthOrderStatisticComputation()

12