Home
last modified time | relevance | path

Searched defs:input_ids (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dtraining_op_helpers.h142 const std::vector<int>& input_ids) { in MaybeLockVariableInputMutexesInOrder()
/external/tensorflow/tensorflow/python/eager/
Dpywrap_tfe_src.cc1577 const std::vector<tensorflow::int64>& input_ids,
1627 std::vector<tensorflow::int64> input_ids = MakeTensorIDList(input_tensors); local
2076 std::vector<tensorflow::int64> input_ids = MakeTensorIDList(inputs); local
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emitter_unnested.cc3349 std::vector<int64> input_ids) { in FilterInputsForShmemTranspose()