Home
last modified time | relevance | path

Searched refs:CreateScalarAddComputation (Results 1 – 25 of 31) 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()
207 XlaComputation add_f32 = CreateScalarAddComputation(F32, &builder); in RunR2ToR0Test()
235 XlaComputation add_f32 = CreateScalarAddComputation(F32, &builder); in RunR2ToR1Test()
446 XlaComputation add_f32 = CreateScalarAddComputation(F32, &builder); in XLA_TEST_F()
477 XlaComputation add_f32 = CreateScalarAddComputation(F32, &builder); in XLA_TEST_F()
509 XlaComputation add_f32 = CreateScalarAddComputation(F32, &builder); in XLA_TEST_F()
525 XlaComputation add_f32 = CreateScalarAddComputation(F32, &builder); in XLA_TEST_F()
571 auto add = CreateScalarAddComputation(F32, &builder); in XLA_TEST_F()
651 auto add = CreateScalarAddComputation(F32, &builder); in XLA_TEST_F()
662 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.cc77 CreateScalarAddComputation(FloatType(), &builder_), in ReduceWindowAdd()
113 CreateScalarAddComputation(FloatType(), &builder_), in XLA_TEST_P()
128 ReduceWindow(input, init, CreateScalarAddComputation(FloatType(), &builder_), in XLA_TEST_P()
650 ? CreateScalarAddComputation(FloatType(), &b) in DoIt()
1078 ? CreateScalarAddComputation(FloatType(), &b) in XLA_TEST_P()
1309 ? CreateScalarAddComputation(FloatType(), &b) in DoIt()
1534 ? CreateScalarAddComputation(FloatType(), &b) in XLA_TEST_P()
Dmap_test.cc357 Map(&builder, {param0, param1}, CreateScalarAddComputation(F32, &builder), in TEST_F()
381 Map(&builder, {param0, param1}, CreateScalarAddComputation(S32, &builder), in XLA_TEST_F()
407 Map(&builder, {param0, param1}, CreateScalarAddComputation(S32, &builder), in XLA_TEST_F()
Dselect_and_scatter_test.cc58 add_s32_ = CreateScalarAddComputation(S32, &builder_); in SelectAndScatterTest()
60 add_f32_ = CreateScalarAddComputation(F32, &builder_); in SelectAndScatterTest()
Dbatch_normalization_test.cc124 XlaComputation add = CreateScalarAddComputation(F32, &builder); in XLA_TEST_F()
138 XlaComputation add = CreateScalarAddComputation(F32, &builder); in XLA_TEST_F()
164 XlaComputation add = CreateScalarAddComputation(F32, &builder); in XLA_TEST_F()
Dvector_ops_simple_test.cc155 auto add = CreateScalarAddComputation(F32, &builder); in XLA_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.cc191 xla::CreateScalarAddComputation(xla::S32, ctx->builder())); in CompileWhereWithSort()
309 xla::CreateScalarAddComputation(S32, b)); in CompileWhereWithPrefixSum()
Dresampler_ops.cc191 xla::CreateScalarAddComputation(xla_type, ctx->builder()), in ScatterToGradData()
467 xla::CreateScalarAddComputation(data_type, ctx->builder()), in CalculateGradWarp()
478 xla::CreateScalarAddComputation(data_type, ctx->builder()), in CalculateGradWarp()
Dunique_op.cc70 auto reducer = xla::CreateScalarAddComputation(xla::S32, ctx->builder()); in CumSumR1()
225 xla::CreateScalarAddComputation(xla::S32, ctx->builder())); in CompileWithAxis()
Ddynamic_partition_op.cc58 xla::CreateScalarAddComputation(xla::S32, ctx->builder())); in CountS32()
/external/tensorflow/tensorflow/compiler/xla/client/lib/
Dsvd.cc135 CreateScalarAddComputation(x_shape.element_type(), builder), in HouseRow()
200 CreateScalarAddComputation(x_shape.element_type(), builder), in HouseCol()
516 CreateScalarAddComputation(d_shape.element_type(), builder), in OneSidedJacobiUpdate()
526 CreateScalarAddComputation(d_shape.element_type(), builder), in OneSidedJacobiUpdate()
542 CreateScalarAddComputation(d_shape.element_type(), builder), in OneSidedJacobiUpdate()
552 CreateScalarAddComputation(d_shape.element_type(), builder), in OneSidedJacobiUpdate()
778 CreateScalarAddComputation(shape.element_type(), builder), in SortBySingularValuesAndPostProcessing()
790 CreateScalarAddComputation(shape.element_type(), builder), in SortBySingularValuesAndPostProcessing()
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.cc49 XlaComputation CreateScalarAddComputation(PrimitiveType type, in CreateScalarAddComputation() function
96 : CreateScalarAddComputation(type, builder); in CreateScalarIdentityWithZeroComputation()
Darithmetic.h35 XlaComputation CreateScalarAddComputation(PrimitiveType type,
Dsvd_test.cc106 CreateScalarAddComputation(F32, builder)) / in GetAverageAbsoluteError()
Dself_adjoint_eig_test.cc115 CreateScalarAddComputation(F32, builder)) / in GetAverageAbsoluteError()
Dmatrix.cc534 CreateScalarAddComputation(x_shape.element_type(), builder), in Einsum()
540 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.cc135 CreateScalarAddComputation( in House()
151 CreateScalarAddComputation(type, builder), {minor_dim}); in House()
Deigh_expander.cc290 zero, CreateScalarAddComputation(norm_type, builder), in ComputeFrobeniusNorms()
295 zero, CreateScalarAddComputation(norm_type, builder), in ComputeFrobeniusNorms()
/external/tensorflow/tensorflow/core/tpu/kernels/
Dtopk_ops.cc82 xla::XlaComputation add = CreateScalarAddComputation(xla::S32, builder); in CreateKthOrderStatisticComputation()

12