Home
last modified time | relevance | path

Searched refs:input_dtypes (Results 1 – 21 of 21) sorted by relevance

/external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/
Dtf_mlir_translate.cc47 const std::vector<std::string>& input_dtypes, in GraphdefToMlirImport() argument
69 TF_RETURN_IF_ERROR(ParseInputArrayInfo(input_arrays, input_dtypes, in GraphdefToMlirImport()
105 const std::vector<std::string>& input_dtypes, in GraphdefToMlirTranslateFunction() argument
113 input, debug_info_file, input_arrays, input_dtypes, input_shapes, in GraphdefToMlirTranslateFunction()
125 absl::string_view input_arrays, absl::string_view input_dtypes, in GraphdefToMlirTranslateFunction() argument
136 TF_RETURN_IF_ERROR(ParseNodeDataTypes(input_dtypes, input_dtype_vector)); in GraphdefToMlirTranslateFunction()
219 const std::vector<std::string>& input_dtypes, in GraphdefToSplattedMlirTranslateFunction() argument
227 input, debug_info_file, input_arrays, input_dtypes, input_shapes, in GraphdefToSplattedMlirTranslateFunction()
269 absl::string_view input_arrays, absl::string_view input_dtypes, in GraphdefToSplattedMlirTranslateFunction() argument
280 TF_RETURN_IF_ERROR(ParseNodeDataTypes(input_dtypes, input_dtype_vector)); in GraphdefToSplattedMlirTranslateFunction()
Dtf_mlir_translate.h43 const std::vector<std::string>& input_dtypes,
59 absl::string_view input_arrays, absl::string_view input_dtypes,
71 const std::vector<std::string>& input_dtypes,
86 absl::string_view input_arrays, absl::string_view input_dtypes,
Dtf_mlir_translate_registration.cc47 input, debug_info_file, input_arrays, input_dtypes, input_shapes, in GraphdefToMlirTranslateFunction()
61 input, debug_info_file, input_arrays, input_dtypes, input_shapes, in GraphdefToSplattedMlirTranslateFunction()
Dtf_mlir_translate_cl.h30 extern llvm::cl::opt<std::string> input_dtypes;
Dtf_mlir_translate_cl.cc30 opt<std::string> input_dtypes( variable
/external/tensorflow/tensorflow/compiler/mlir/lite/
Dtf_to_tfl_flatbuffer.cc101 absl::string_view input_arrays, absl::string_view input_dtypes, in LoadFromGraphdefOrMlirSource() argument
123 file->getBuffer(), debug_info_file, input_arrays, input_dtypes, in LoadFromGraphdefOrMlirSource()
130 file->getBuffer(), debug_info_file, input_arrays, input_dtypes, in LoadFromGraphdefOrMlirSource()
Dtf_to_tfl_flatbuffer.h43 absl::string_view input_arrays, absl::string_view input_dtypes,
Dtf_tfl_translate.cc182 specs, debug_info_file, input_arrays, input_dtypes, input_shapes, in main()
/external/tensorflow/tensorflow/compiler/mlir/tfjs/translate/
Dtf_to_tfjs_json.cc76 absl::string_view input_dtypes, absl::string_view input_shapes, in LoadFromGraphdefOrMlirSource() argument
95 file->getBuffer(), debug_info_file, input_arrays, input_dtypes, in LoadFromGraphdefOrMlirSource()
Dtf_to_tfjs_json.h42 absl::string_view input_dtypes, absl::string_view input_shapes,
Dtf_tfjs_translate.cc143 input_arrays, input_dtypes, input_shapes, output_arrays, in main()
/external/tensorflow/tensorflow/core/common_runtime/eager/
Dkernel_and_device.h170 virtual const DataTypeVector& input_dtypes() const = 0;
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
Dexecute.cc194 const DataType* input_types = &kernel->input_dtypes()[0]; in ValidateInputTypeAndPlacement()
/external/tensorflow/tensorflow/c/eager/
Dtape.h160 gtl::ArraySlice<tensorflow::DataType> input_dtypes,
285 gtl::ArraySlice<tensorflow::DataType> input_dtypes,
409 gtl::ArraySlice<tensorflow::DataType> input_dtypes, in RecordOperation() argument
412 if (!ShouldRecord(input_tensor_id, input_dtypes)) { in RecordOperation()
1026 gtl::ArraySlice<tensorflow::DataType> input_dtypes, in Accumulate() argument
1042 op_type, output_tensors, input_tensor_id, input_dtypes, in Accumulate()
1046 if (!ShouldRecord(input_tensor_id, input_dtypes)) { in Accumulate()
Dgradients.cc230 std::vector<tensorflow::DataType> input_dtypes(inputs.size()); in RecordOperation() local
233 input_dtypes[i] = inputs[i]->DataType(); in RecordOperation()
240 op_name, tape_tensors, input_ids, input_dtypes, in RecordOperation()
/external/tensorflow/tensorflow/python/eager/
Dpywrap_tfe_src.cc2295 const std::vector<tensorflow::DataType>& input_dtypes, argument
2304 tape->tape->RecordOperation(op_type, output_info, input_ids, input_dtypes,
2315 const std::vector<tensorflow::DataType>& input_dtypes, argument
2381 op_type, input_info, output_info, input_ids, input_dtypes,
2516 const std::vector<tensorflow::DataType>& input_dtypes, argument
2534 input_dtypes, backward_function_getter, backward_function_killer,
2539 TapeSetRecordBackprop(op_type_str, output_info, input_ids, input_dtypes,
2557 std::vector<tensorflow::DataType> input_dtypes = local
2579 op_type, input_tensors, output_tensors, input_ids, input_dtypes,
2593 op_type, input_tensors, output_tensors, input_ids, input_dtypes,
[all …]
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/
Dtf_xla_mlir_translate.cc335 mlir::StringRefToView(input_shapes), mlir::StringRefToView(input_dtypes), in MlirTfGraphToHloTextTranslateFunction()
/external/tensorflow/tensorflow/compiler/jit/
Dextract_outside_compilation_pass.cc357 std::vector<DataType> input_dtypes; in BuildXlaHostComputeNodeDef() local
358 TF_RETURN_IF_ERROR(GetNodeAttr(call_node->attrs(), "Tinputs", &input_dtypes)); in BuildXlaHostComputeNodeDef()
359 std::vector<NodeDefBuilder::NodeOut> inputs(input_dtypes.size()); in BuildXlaHostComputeNodeDef()
365 const int input_dtypes_size = input_dtypes.size(); in BuildXlaHostComputeNodeDef()
370 e->src()->name(), e->src_output(), input_dtypes[e->dst_input()]}; in BuildXlaHostComputeNodeDef()
/external/tensorflow/tensorflow/python/keras/engine/
Dtraining_utils_v1.py1414 input_dtypes = nest.map_structure(lambda t: t.dtype, model.inputs)
1415 return nest.map_structure(math_ops.cast, x, input_dtypes)
Dbase_layer_v1.py609 input_dtypes = [s.dtype for s in nest.flatten(input_signature)]
612 dtype = input_dtypes[0]
Dbase_layer.py819 input_dtypes = [s.dtype for s in nest.flatten(input_signature)]
822 dtype = input_dtypes[0]