Searched refs:arg_vector (Results 1 - 2 of 2) sorted by relevance
/external/tensorflow/tensorflow/compiler/xla/tests/ |
H A D | array_elementwise_ops_test.cc | 1881 auto arg_vector = builder.ConstantR1<float>({2.0f, 10.0f, -5.0f, 1.0f, 4.0f}); local 1886 builder.Add(builder.Clamp(min_vector, arg_vector, max_scalar), 1887 builder.Clamp(min_scalar, arg_vector, max_vector)), 1888 builder.Add(builder.Clamp(min_vector, arg_vector, max_vector), 1889 builder.Clamp(min_scalar, arg_vector, max_scalar))); 1898 auto arg_vector = builder.ConstantR1<int32>({2, 10, -5, 1, 4, 10}); local 1900 auto clamp = builder.Clamp(min_vector, arg_vector, max_vector); 1909 auto arg_vector = builder.ConstantR1<int32>({2, 10, -5, 1, 4}); local 1914 builder.Add(builder.Clamp(min_vector, arg_vector, max_scalar), 1915 builder.Clamp(min_scalar, arg_vector, max_vecto 1925 auto arg_vector = builder.ConstantR1<uint32>({2, 10, 5, 1, 4, 10}); local 1936 auto arg_vector = builder.ConstantR1<uint32>({2, 10, 0, 1, 4}); local [all...] |
H A D | local_client_execute_test.cc | 553 std::vector<std::unique_ptr<Literal>> arg_vector; local 554 arg_vector.push_back(std::move(arg_literal)); 555 arg_literal = Literal::MakeTupleOwned(std::move(arg_vector));
|
Completed in 98 milliseconds