Searched defs:arg_vector (Results 1 - 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/tests/
H A Dlocal_client_execute_test.cc553 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));
H A Darray_elementwise_ops_test.cc1881 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...]

Completed in 124 milliseconds