Home
last modified time | relevance | path

Searched refs:ConstantLiteral (Results 1 – 25 of 27) 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.cc592 Reduce(ConstantLiteral(&builder, input_literal), in XLA_TEST_F()
607 Reduce(ConstantLiteral(&builder, input_literal), in XLA_TEST_F()
624 Reduce(ConstantLiteral(&builder, input_literal), initial_value, min, {0, 1}); in XLA_TEST_F()
636 Reduce(ConstantLiteral(&builder, input_literal), initial_value, max, {0, 1}); in XLA_TEST_F()
643 auto m = ConstantLiteral(&builder, literal_2d_); in XLA_TEST_F()
654 auto m = ConstantLiteral(&builder, literal_2d_); in XLA_TEST_F()
664 auto m = ConstantLiteral(&builder, literal_2d_); in XLA_TEST_F()
674 auto m = ConstantLiteral(&builder, literal_3d_); in XLA_TEST_F()
684 auto m = ConstantLiteral(&builder, literal_3d_); in XLA_TEST_F()
694 auto m = ConstantLiteral(&builder, literal_3d_); in XLA_TEST_F()
[all …]
Dbatch_normalization_test.cc106 auto x = ConstantLiteral(&builder, input_literal_); in XLA_TEST_P()
122 auto x = ConstantLiteral(&builder, input_literal_); in XLA_TEST_P()
139 auto input_activations = ConstantLiteral(&builder, input_literal_); in XLA_TEST_P()
150 auto input_activations = ConstantLiteral(&builder, input_literal_); in XLA_TEST_P()
176 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()
Dmatrix.cc87 return ConstantLiteral(builder, Literal{zero_size_shape}); in GetMatrixDiagonal()
/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()
122 return xla::ConstantLiteral(builder, literal); in IntegerLiteral()
/external/tensorflow/tensorflow/compiler/xla/client/
Dxla_builder.h376 XlaOp ConstantLiteral(const LiteralSlice& literal);
751 friend XlaOp ConstantLiteral(XlaBuilder* builder,
1117 XlaOp ConstantLiteral(XlaBuilder* builder, const LiteralSlice& literal);
1992 return ConstantLiteral(builder, LiteralUtil::CreateR0<NativeT>(value)); in ConstantR0()
2001 return ConstantLiteral(builder, literal); in ConstantR1()
2009 return ConstantLiteral(builder, literal); in ConstantR1()
2014 return ConstantLiteral(builder, LiteralUtil::CreateR1(values)); in ConstantR1()
2020 return ConstantLiteral(builder, LiteralUtil::CreateR2<NativeT>(values)); in ConstantR2()
2027 return ConstantLiteral( in ConstantFromArrayWithLayout()
2033 return ConstantLiteral(builder, in ConstantFromArray()
[all …]
Dxla_builder.cc681 XlaOp XlaBuilder::ConstantLiteral(const LiteralSlice& literal) { in ConstantLiteral() function in xla::XlaBuilder
2890 XlaOp ConstantLiteral(XlaBuilder* builder, const LiteralSlice& literal) { in ConstantLiteral() function
2891 return builder->ConstantLiteral(literal); in ConstantLiteral()
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dtensor_list_utils.cc193 xla::ConstantLiteral(b, xla::LiteralUtil::Zero(xla::PrimitiveType::S32)); in BuildUninitializedTensorList()
259 xla::ConstantLiteral(b, xla::LiteralUtil::Zero(shape.element_type())); in CreateZerosTensorListWithShape()
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/xla/python/
Dxla_client.py826 return ops.ConstantLiteral(self._builder, value)
Dxla.cc845 ops.def("ConstantLiteral", &ConstantLiteral); in PYBIND11_MODULE()
/external/tensorflow/tensorflow/compiler/mlir/xla/
Dmlir_hlo_to_hlo.cc855 xla::ConstantLiteral(builder, literal_or.ValueOrDie()); in Lower()

12