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.cc2097 auto arg_vector = in XLA_TEST_F() local
2103 Add(Add(Clamp(min_vector, arg_vector, max_scalar), in XLA_TEST_F()
2104 Clamp(min_scalar, arg_vector, max_vector)), in XLA_TEST_F()
2105 Add(Clamp(min_vector, arg_vector, max_vector), in XLA_TEST_F()
2106 Clamp(min_scalar, arg_vector, max_scalar))); in XLA_TEST_F()
2115 auto arg_vector = ConstantR1<int32>(&builder, {2, 10, -5, 1, 4, 10}); in XLA_TEST_F() local
2117 Clamp(min_vector, arg_vector, max_vector); in XLA_TEST_F()
2126 auto arg_vector = ConstantR1<int32>(&builder, {2, 10, -5, 1, 4}); in XLA_TEST_F() local
2130 Add(Add(Clamp(min_vector, arg_vector, max_scalar), in XLA_TEST_F()
2131 Clamp(min_scalar, arg_vector, max_vector)), in XLA_TEST_F()
[all …]
Dlocal_client_execute_test.cc528 std::vector<Literal> arg_vector; in XLA_TEST_F() local
529 arg_vector.push_back(std::move(arg_literal)); in XLA_TEST_F()
530 arg_literal = LiteralUtil::MakeTupleOwned(std::move(arg_vector)); in XLA_TEST_F()
/external/tensorflow/tensorflow/compiler/tf2xla/
Dxla_compiler.cc550 const std::vector<XlaCompiler::Argument> arg_vector(args.begin(), args.end()); in CompileFunction() local
551 auto it = cache_.find({function_id, arg_vector}); in CompileFunction()
607 cache_[{function_id, arg_vector}] = *result; in CompileFunction()