| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/ |
| D | tf_mlir_translate.cc | 47 const std::vector<std::string>& input_dtypes, in GraphdefToMlirImport()
107 const std::vector<std::string>& input_dtypes, in GraphdefToMlirTranslateFunction()
127 absl::string_view input_arrays, absl::string_view input_dtypes, in GraphdefToMlirTranslateFunction()
245 const std::vector<std::string>& input_dtypes, in GraphdefToSplattedMlirTranslateFunction()
296 absl::string_view input_arrays, absl::string_view input_dtypes, in GraphdefToSplattedMlirTranslateFunction()
|
| D | tf_mlir_translate_cl.cc | 31 opt<std::string> input_dtypes( variable
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | batch_kernel_test_util.cc | 36 std::vector<DataType> input_dtypes({DataType::DT_INT64, DataType::DT_INT64}); in Init() local
|
| /external/tensorflow/tensorflow/cc/gradients/ |
| D | functional_grad.cc | 37 std::vector<DataType> input_dtypes; in PartitionedCallGrad() local
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | node_file_writer.cc | 138 AttrValue& input_dtypes = (*def.mutable_attr())["_input_dtypes"]; in RecordNodeExecution() local
|
| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | kernel_and_device.h | 241 const DataTypeVector& input_dtypes() const override { in input_dtypes() function
339 const DataTypeVector& input_dtypes() const override { return input_dtypes_; } in input_dtypes() function
|
| /external/tensorflow/tensorflow/core/kernels/batching_util/ |
| D | batch_input_task_test.cc | 87 std::vector<DataType> input_dtypes({DataType::DT_INT64, DataType::DT_INT64}); in CreateBatchKernelNodeDef() local
|
| /external/tensorflow/tensorflow/compiler/mlir/tfjs/translate/ |
| D | tf_to_tfjs_json.cc | 77 absl::string_view input_dtypes, absl::string_view input_shapes, in LoadFromGraphdefOrMlirSource()
|
| /external/tensorflow/tensorflow/core/tfrt/utils/ |
| D | graph_partition.cc | 228 std::vector<DataType> input_dtypes(input_nodes.size()); in BuildPartitionedCallOp() local
305 std::vector<DataType> input_dtypes(num_output_nodes); in BuildStatefulPartitionedCallOp() local
|
| D | tfrt_graph_execution_state_test.cc | 1035 std::vector<DataType> input_dtypes; in MakeOuterGraph() local
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/ |
| D | tf_to_tfl_flatbuffer.cc | 111 absl::string_view input_arrays, absl::string_view input_dtypes, in LoadFromGraphdefOrMlirSource()
|
| /external/tensorflow/tensorflow/c/eager/ |
| D | tape.h | 409 gtl::ArraySlice<tensorflow::DataType> input_dtypes, in RecordOperation()
1026 gtl::ArraySlice<tensorflow::DataType> input_dtypes, in Accumulate()
|
| D | gradients.cc | 230 std::vector<tensorflow::DataType> input_dtypes(inputs.size()); in RecordOperation() local
|
| /external/tensorflow/tensorflow/python/eager/ |
| D | pywrap_tfe_src.cc | 2367 const std::vector<tensorflow::DataType>& input_dtypes,
2387 const std::vector<tensorflow::DataType>& input_dtypes,
2588 const std::vector<tensorflow::DataType>& input_dtypes,
2629 std::vector<tensorflow::DataType> input_dtypes = local
2683 std::vector<tensorflow::DataType> input_dtypes = local
2734 std::vector<tensorflow::DataType> input_dtypes = local
3229 std::vector<tensorflow::DataType> input_dtypes = MakeTensorDtypeList(inputs); local
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | extract_outside_compilation_pass.cc | 356 std::vector<DataType> input_dtypes; in BuildXlaHostComputeNodeDef() local
|