HomeSort by relevance Sort by last modified time
    Searched defs:key_arg (Results 1 - 2 of 2) sorted by null

  /external/tensorflow/tensorflow/compiler/xla/tests/
test_utils_test.cc 159 const Literal& key_arg = args[0]; local
162 for (const float& value : key_arg.data<float>()) {
190 const Literal& key_arg = args[0]; local
193 for (const int32& value : key_arg.data<int32>()) {
221 const Literal& key_arg = args[0]; local
224 for (const bfloat16& value : key_arg.data<bfloat16>()) {
  /external/tensorflow/tensorflow/compiler/jit/
extract_outside_compilation_pass.cc 857 NodeDefBuilder arg_builder("key_arg", FunctionLibraryDefinition::kArgOp)
1044 Node* key_arg = nullptr; local
1124 Node* key_arg = nullptr; local
    [all...]

Completed in 340 milliseconds