Searched defs:input_dtypes (Results 1 – 9 of 9) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/ |
| D | tf_mlir_translate.cc | 47 const std::vector<std::string>& input_dtypes, in GraphdefToMlirImport()
105 const std::vector<std::string>& input_dtypes, in GraphdefToMlirTranslateFunction()
125 absl::string_view input_arrays, absl::string_view input_dtypes, in GraphdefToMlirTranslateFunction()
219 const std::vector<std::string>& input_dtypes, in GraphdefToSplattedMlirTranslateFunction()
269 absl::string_view input_arrays, absl::string_view input_dtypes, in GraphdefToSplattedMlirTranslateFunction()
|
| D | tf_mlir_translate_cl.cc | 30 opt<std::string> input_dtypes( variable
|
| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | kernel_and_device.h | 232 const DataTypeVector& input_dtypes() const override { in input_dtypes() function
317 const DataTypeVector& input_dtypes() const override { return input_dtypes_; } in input_dtypes() function
|
| /external/tensorflow/tensorflow/compiler/mlir/tfjs/translate/ |
| D | tf_to_tfjs_json.cc | 76 absl::string_view input_dtypes, absl::string_view input_shapes, in LoadFromGraphdefOrMlirSource()
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/ |
| D | tf_to_tfl_flatbuffer.cc | 101 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 | 2295 const std::vector<tensorflow::DataType>& input_dtypes,
2315 const std::vector<tensorflow::DataType>& input_dtypes,
2516 const std::vector<tensorflow::DataType>& input_dtypes,
2557 std::vector<tensorflow::DataType> input_dtypes = local
2611 std::vector<tensorflow::DataType> input_dtypes = local
2662 std::vector<tensorflow::DataType> input_dtypes = local
3150 std::vector<tensorflow::DataType> input_dtypes = MakeTensorDtypeList(inputs); local
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | extract_outside_compilation_pass.cc | 357 std::vector<DataType> input_dtypes; in BuildXlaHostComputeNodeDef() local
|