Searched defs:input_dtypes (Results 1 – 11 of 11) 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()
235 const std::vector<std::string>& input_dtypes, in GraphdefToSplattedMlirTranslateFunction()
285 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/cc/gradients/ |
| D | functional_grad.cc | 37 std::vector<DataType> input_dtypes; in PartitionedCallGrad() local
|
| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | kernel_and_device.h | 236 const DataTypeVector& input_dtypes() const override { in input_dtypes() function
324 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/core/kernels/batching_util/ |
| D | batch_input_task_test.cc | 108 std::vector<DataType> input_dtypes({DataType::DT_INT64, DataType::DT_INT64}); in CreateBatchKernelNodeDef() local
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/ |
| D | tf_to_tfl_flatbuffer.cc | 105 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 | 2307 const std::vector<tensorflow::DataType>& input_dtypes,
2327 const std::vector<tensorflow::DataType>& input_dtypes,
2528 const std::vector<tensorflow::DataType>& input_dtypes,
2569 std::vector<tensorflow::DataType> input_dtypes = local
2623 std::vector<tensorflow::DataType> input_dtypes = local
2674 std::vector<tensorflow::DataType> input_dtypes = local
3162 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
|