Home
last modified time | relevance | path

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

12

/external/tensorflow/tensorflow/compiler/xla/tests/
Dbroadcast_simple_test.cc293 ConstantLiteral(&b, LiteralUtil::CreateR3<float>( in XLA_TEST_F()
400 auto r1 = ConstantLiteral(&b, LiteralUtil::CreateR3<float>({{{1, 2}}})); in XLA_TEST_F()
401 auto r3 = ConstantLiteral( in XLA_TEST_F()
413 auto r1 = ConstantLiteral(&b, LiteralUtil::CreateR3<float>({{{1}, {2}}})); in XLA_TEST_F()
414 auto r3 = ConstantLiteral( in XLA_TEST_F()
427 ConstantLiteral(&b, LiteralUtil::CreateR3<float>({{{1, 2}, {3, 4}}})); in XLA_TEST_F()
428 auto r3 = ConstantLiteral( in XLA_TEST_F()
441 ConstantLiteral(&b, LiteralUtil::CreateR3<float>({{{1, 2}}, {{3, 4}}})); in XLA_TEST_F()
442 auto r3 = ConstantLiteral( in XLA_TEST_F()
454 auto r1 = ConstantLiteral( in XLA_TEST_F()
[all …]
Dcall_test.cc80 auto constant = ConstantLiteral(&builder, LiteralUtil::CreateR0<float>(42.0)); in XLA_TEST_F()
89 auto x = ConstantLiteral(&builder, LiteralUtil::CreateR1<float>({})); in XLA_TEST_F()
90 auto y = ConstantLiteral(&builder, LiteralUtil::CreateR1<float>({})); in XLA_TEST_F()
100 ConstantLiteral(&builder, LiteralUtil::CreateR1<float>({1.0f, 2.0f})); in XLA_TEST_F()
102 ConstantLiteral(&builder, LiteralUtil::CreateR1<float>({2.0f, 3.0f})); in XLA_TEST_F()
144 Call(&builder, callee, {ConstantLiteral(&builder, elem)}); in XLA_TEST_F()
Dconstants_test.cc116 ConstantLiteral(&builder, LiteralUtil::CreateR3FromArray3D<float>( in TEST_F()
132 ConstantLiteral(&builder, LiteralUtil::CreateR3FromArray3D<float>(array3d)); in TEST_F()
150 ConstantLiteral(&builder, input_literal); in TEST_F()
164 ConstantLiteral(&builder, LiteralUtil::MakeTupleFromSlices( in TEST_F()
178 ConstantLiteral(&builder, LiteralUtil::CreateToken()); in TEST_F()
Ddynamic_ops_test.cc141 auto input = ConstantLiteral(&builder, input_values); in RunR1()
171 auto input = ConstantLiteral(&builder, input_values); in RunR2()
205 auto input = ConstantLiteral(&builder, input_values); in RunR3()
393 auto input = ConstantLiteral(&builder, input_value); in RunR0()
394 auto update = ConstantLiteral(&builder, update_value); in RunR0()
424 auto input = ConstantLiteral(&builder, input_values); in RunR1()
425 auto update = ConstantLiteral(&builder, update_values); in RunR1()
458 auto input = ConstantLiteral(&builder, input_values); in RunR2()
459 auto update = ConstantLiteral(&builder, update_values); in RunR2()
498 auto input = ConstantLiteral(&builder, input_values); in RunR3()
[all …]
Dreduce_test.cc591 Reduce(ConstantLiteral(&builder, input_literal), in XLA_TEST_F()
606 Reduce(ConstantLiteral(&builder, input_literal), in XLA_TEST_F()
623 Reduce(ConstantLiteral(&builder, input_literal), initial_value, min, {0, 1}); in XLA_TEST_F()
635 Reduce(ConstantLiteral(&builder, input_literal), initial_value, max, {0, 1}); in XLA_TEST_F()
642 auto m = ConstantLiteral(&builder, literal_2d_); in XLA_TEST_F()
653 auto m = ConstantLiteral(&builder, literal_2d_); in XLA_TEST_F()
663 auto m = ConstantLiteral(&builder, literal_2d_); in XLA_TEST_F()
673 auto m = ConstantLiteral(&builder, literal_3d_); in XLA_TEST_F()
683 auto m = ConstantLiteral(&builder, literal_3d_); in XLA_TEST_F()
693 auto m = ConstantLiteral(&builder, literal_3d_); in XLA_TEST_F()
[all …]
Dbatch_normalization_test.cc105 auto x = ConstantLiteral(&builder, input_literal_); in XLA_TEST_P()
121 auto x = ConstantLiteral(&builder, input_literal_); in XLA_TEST_P()
138 auto input_activations = ConstantLiteral(&builder, input_literal_); in XLA_TEST_P()
149 auto input_activations = ConstantLiteral(&builder, input_literal_); in XLA_TEST_P()
175 CheckShape(&builder, ConstantLiteral(&builder, input_literal_), in XLA_TEST_P()
Dmultiple_devices_on_host_test.cc32 ConstantLiteral(&b, LiteralUtil::CreateR0<int32>(1)), in BuildComputation()
Dtuple_test.cc59 ConstantLiteral(&builder, value); in XLA_TEST_F()
73 ConstantLiteral(&builder, value); in XLA_TEST_F()
Dconvolution_variants_test.cc1339 auto gradients = ConstantLiteral(&builder, gradients_literal); in XLA_TEST_F()
1344 auto weights = ConstantLiteral(&builder, weights_literal); in XLA_TEST_F()
1363 auto activations = ConstantLiteral(&builder, activations_literal); in XLA_TEST_F()
1368 auto gradients = ConstantLiteral(&builder, gradients_literal); in XLA_TEST_F()
Dclient_library_test_base.cc603 return ConstantLiteral(builder, use_bfloat16_ in CreateConstantFromLiteral()
/external/tensorflow/tensorflow/compiler/xla/client/lib/
Dconstants.cc24 return ConstantLiteral(builder, LiteralUtil::Zero(type)); in Zero()
41 return ConstantLiteral(builder, LiteralUtil::One(type)); in One()
64 return ConstantLiteral(builder, LiteralUtil::MinValue(type)); in MinValue()
102 return ConstantLiteral(builder, LiteralUtil::MaxValue(type)); in MaxValue()
Dloops.cc101 ConstantLiteral(body_builder, LiteralUtil::One(num_iterations_type)))); in ForEachIndex()
114 ConstantLiteral(builder, LiteralUtil::Zero(num_iterations_type))); in ForEachIndex()
Dtesting.cc52 ConstantLiteral(builder, LiteralUtil::One(shape.element_type())), in BuildFakeDataOpOnDevice()
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dindex_ops_cpu.cc94 args.push_back(xla::ConstantLiteral( in Compile()
99 args.push_back(xla::ConstantLiteral( in Compile()
102 xla::ConstantLiteral(&b, xla::LiteralUtil::CreateR0<int32>(axis))); in Compile()
Dtensor_list_utils.cc92 xla::ConstantLiteral(builder, xla::LiteralUtil::Zero( in InitializeTensorList()
Ddynamic_stitch_op.cc126 ctx->SetOutput(0, xla::ConstantLiteral(ctx->builder(), empty_literal)); in Compile()
Dresampler_ops.cc301 auto grad_data = xla::ConstantLiteral( in CalculateGradData()
/external/tensorflow/tensorflow/compiler/tf2xla/
Dxla_helpers.cc40 return xla::ConstantLiteral(b, xla::LiteralUtil::Zero(type)); in Zero()
46 return xla::ConstantLiteral(b, xla::LiteralUtil::One(type)); in One()
Dxla_expression.cc87 return xla::ConstantLiteral(builder, literal); in AsXlaOp()
/external/tensorflow/tensorflow/compiler/tf2xla/lib/
Dutil.cc34 xla::ConstantLiteral(builder, in Zeros()
119 return xla::ConstantLiteral(builder, literal); in IntegerLiteral()
/external/tensorflow/tensorflow/compiler/xla/client/
Dxla_builder.h327 XlaOp ConstantLiteral(const LiteralSlice& literal);
712 friend XlaOp ConstantLiteral(XlaBuilder* builder,
1050 XlaOp ConstantLiteral(XlaBuilder* builder, const LiteralSlice& literal);
1926 return ConstantLiteral(builder, LiteralUtil::CreateR0<NativeT>(value)); in ConstantR0()
1931 return ConstantLiteral(builder, LiteralUtil::CreateR1<NativeT>(values)); in ConstantR1()
1939 return ConstantLiteral(builder, literal); in ConstantR1()
1944 return ConstantLiteral(builder, LiteralUtil::CreateR1(values)); in ConstantR1()
1950 return ConstantLiteral(builder, LiteralUtil::CreateR2<NativeT>(values)); in ConstantR2()
1957 return ConstantLiteral( in ConstantFromArrayWithLayout()
1963 return ConstantLiteral(builder, in ConstantFromArray()
[all …]
Dxla_builder.cc589 XlaOp XlaBuilder::ConstantLiteral(const LiteralSlice& literal) { in ConstantLiteral() function in xla::XlaBuilder
1620 ConstantLiteral(value.builder(), LiteralUtil::Zero(signed_type))); in BitcastConvertFloatingPointToIntegral()
2835 XlaOp ConstantLiteral(XlaBuilder* builder, const LiteralSlice& literal) { in ConstantLiteral() function
2836 return builder->ConstantLiteral(literal); in ConstantLiteral()
/external/tensorflow/tensorflow/compiler/xla/python/
Dlocal_computation_builder.cc457 LocalOp ComputationBuilder::ConstantLiteral(const Literal& literal) { in ConstantLiteral() function in xla::swig::ComputationBuilder
458 return xla::ConstantLiteral(&builder_, literal); in ConstantLiteral()
Dlocal_computation_builder.h234 LocalOp ConstantLiteral(const Literal& literal);
Dlocal_computation_builder.i296 %unignore xla::swig::ComputationBuilder::ConstantLiteral;

12