Home
last modified time | relevance | path

Searched refs:arg_vector (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/tests/
Darray_elementwise_ops_test.cc2203 auto arg_vector = in XLA_TEST_F() local
2209 Add(Add(Clamp(min_vector, arg_vector, max_scalar), in XLA_TEST_F()
2210 Clamp(min_scalar, arg_vector, max_vector)), in XLA_TEST_F()
2211 Add(Clamp(min_vector, arg_vector, max_vector), in XLA_TEST_F()
2212 Clamp(min_scalar, arg_vector, max_scalar))); in XLA_TEST_F()
2221 auto arg_vector = ConstantR1<int32>(&builder, {2, 10, -5, 1, 4, 10}); in XLA_TEST_F() local
2223 Clamp(min_vector, arg_vector, max_vector); in XLA_TEST_F()
2232 auto arg_vector = ConstantR1<int32>(&builder, {2, 10, -5, 1, 4}); in XLA_TEST_F() local
2236 Add(Add(Clamp(min_vector, arg_vector, max_scalar), in XLA_TEST_F()
2237 Clamp(min_scalar, arg_vector, max_vector)), in XLA_TEST_F()
[all …]
Dlocal_client_execute_test.cc531 std::vector<Literal> arg_vector; in XLA_TEST_F() local
532 arg_vector.push_back(std::move(arg_literal)); in XLA_TEST_F()
533 arg_literal = LiteralUtil::MakeTupleOwned(std::move(arg_vector)); in XLA_TEST_F()
/external/tensorflow/tensorflow/compiler/tf2xla/
Dxla_compiler.cc729 const std::vector<XlaCompiler::Argument> arg_vector(args.begin(), args.end()); in CompileFunction() local
730 auto it = cache_.find({function_id, arg_vector}); in CompileFunction()
828 cache_[{function_id, arg_vector}] = *result; in CompileFunction()