Searched refs:input_ids (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | training_op_helpers.h | 142 const std::vector<int>& input_ids) { in MaybeLockVariableInputMutexesInOrder() argument
144 for (auto i : input_ids) { in MaybeLockVariableInputMutexesInOrder()
156 for (auto input : input_ids) { in MaybeLockVariableInputMutexesInOrder()
|
| /external/tensorflow/tensorflow/python/eager/ |
| D | pywrap_tfe_src.cc | 1577 const std::vector<tensorflow::int64>& input_ids, argument
1614 tape->tape->RecordOperation(op_type_str, output_info, input_ids,
1627 std::vector<tensorflow::int64> input_ids = MakeTensorIDList(input_tensors); local
1635 op_type, output_tensors, input_ids, input_dtypes,
2076 std::vector<tensorflow::int64> input_ids = MakeTensorIDList(inputs); local
2083 if (tape->tape->ShouldRecord(input_ids, input_dtypes)) {
2127 op_name, results, input_ids, input_dtypes,
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | ir_emitter_unnested.cc | 3349 std::vector<int64> input_ids) { in FilterInputsForShmemTranspose() argument
3351 for (int64 i = 0; i < input_ids.size(); ++i) { in FilterInputsForShmemTranspose()
3352 const HloInstruction* input = fusion->fused_parameter(input_ids[i]); in FilterInputsForShmemTranspose()
3354 filtered_input_ids.push_back(input_ids[i]); in FilterInputsForShmemTranspose()
|