Searched refs:inputs_list (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/python/distribute/ |
| D | input_ops.py | 91 inputs_list = []
96 inputs_list.append(remapped_input)
98 inputs_list.append(input_tensor_op.outputs[input_tensor.value_index])
102 inputs_list, [o.dtype for o in op_to_clone.outputs],
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | rnn_test.py | 414 inputs_list = [
418 inputs = np.dstack(inputs_list).transpose([0, 2, 1]) # batch x time x depth
424 x, trainable=False).value() for x in inputs_list
462 inputs_list = [
466 inputs = np.dstack(inputs_list).transpose([0, 2, 1]) # batch x time x depth
473 x, trainable=False).value() for x in inputs_list
525 inputs_list = [
535 x, trainable=False).value() for x in inputs_list
547 x, trainable=False).value() for x in inputs_list
593 inputs_list = [
[all …]
|
| /external/tensorflow/tensorflow/stream_executor/gpu/ |
| D | asm_compiler.cc | 379 std::ostringstream inputs_list; in BundleGpuAsm() local
382 inputs_list << "/dev/null"; in BundleGpuAsm()
397 inputs_list << "," << img_path; in BundleGpuAsm()
422 clang_offload_bundler_path, absl::StrCat("--inputs=", inputs_list.str()), in BundleGpuAsm()
|
| /external/tflite-support/tensorflow_lite_support/codegen/ |
| D | android_java_generator.h | 63 std::string inputs_list; member
|
| D | android_java_generator.cc | 190 model_info.inputs_list += ", "; in CreateModelInfo()
193 model_info.inputs_list += model_info.inputs[i].name; in CreateModelInfo()
242 code_writer->SetTokenValue("INPUTS_LIST", model_info.inputs_list); in SetCodeWriterWithModelInfo()
|