Home
last modified time | relevance | path

Searched defs:input_types (Results 1 – 25 of 54) sorted by relevance

123

/external/tensorflow/tensorflow/core/framework/
Dnode_properties_test.cc63 const DataTypeVector& input_types, in ValidateNodeProperties()
85 DataTypeVector input_types{DT_FLOAT, DT_INT32}; in TEST() local
115 DataTypeVector input_types{DT_FLOAT, DT_INT32}; in TEST() local
Dnode_properties.cc29 DataTypeVector input_types; in CreateFromNodeDef() local
Dnode_properties.h55 DataTypeVector input_types; member
/external/tensorflow/tensorflow/lite/kernels/shim/test_op/
Dsimple_tflite_op_test.cc34 const std::vector<tflite::TensorType>& input_types, in SimpleOpModel()
81 std::vector<tflite::TensorType> input_types = {tflite::TensorType_STRING, in TEST() local
114 std::vector<tflite::TensorType> input_types = {tflite::TensorType_STRING}; in TEST() local
/external/tensorflow/tensorflow/compiler/mlir/lite/utils/
Dperception_ops_utils_test.cc37 mlir::Builder* builder, const SmallVector<mlir::Type, NInput>& input_types, in createMaxUnpoolingFunc()
56 SmallVector<mlir::Type, 2> input_types{input_type, indices_type}; in createMaxUnpoolingFunc() local
143 SmallVector<mlir::Type, 1> input_types{input_type}; in TEST_F() local
Dlstm_utils_test.cc60 SmallVector<mlir::Type, 4> input_types{input_type, weight_type, bias_type, in createLstmCompositeFunc() local
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Delemental_ir_emitter.cc82 absl::Span<const PrimitiveType> input_types, PrimitiveType output_type, in EmitDeviceMathCall()
124 absl::Span<const PrimitiveType> input_types, PrimitiveType output_type) { in EmitLlvmIntrinsicMathCall()
147 absl::Span<const PrimitiveType> input_types, PrimitiveType output_type, in EmitMathCall()
Dtarget_util.cc195 absl::Span<const PrimitiveType> input_types, PrimitiveType output_type, in EmitDeviceFunctionCall()
/external/tensorflow/tensorflow/python/framework/
Dfunction.py93 def __init__(self, *input_types, **kwargs):
234 input_types, argument
628 def instantiate(self, input_types): argument
813 input_types=None, argument
Dfunc_graph.py506 input_types=None, argument
539 input_types=None, argument
/external/tensorflow/tensorflow/lite/toco/logging/
Dconversion_log_util.cc173 TFLITE_PROTO_NS::RepeatedPtrField<std::string>* input_types, in GetInputAndOutputTypes()
250 TFLITE_PROTO_NS::RepeatedPtrField<std::string> input_types, output_types; in PopulateConversionLog() local
Dconversion_log_util_test.cc152 TFLITE_PROTO_NS::RepeatedPtrField<std::string> input_types, output_types; in TEST() local
/external/tensorflow/tensorflow/python/ops/
Dwhile_v2.py990 input_types=None, argument
1047 input_types=None, argument
1093 input_types=None, argument
/external/tensorflow/tensorflow/lite/toco/
Dtoco_cmdline_flags.cc334 std::vector<std::string> input_types = in ReadTocoFlagsFromCommandLineFlags() local
/external/tensorflow/tensorflow/tools/graph_transforms/
Dinsert_logging.cc111 DataTypeVector input_types; in InsertLogging() local
/external/vixl/tools/test_generator/
Dparser.py54 def __init__(self, operand_types, operand_variants, input_types, argument
Dgenerator.py252 def GenerateInputs(self, input_types): argument
/external/tensorflow/tensorflow/compiler/mlir/tfr/passes/
Draise_to_tf.cc163 const llvm::SmallVectorImpl<Attribute>& input_types, in CastValuesToSameType()
186 ArrayRef<Attribute> input_types, in AddDerivedAttrs()
/external/tensorflow/tensorflow/core/profiler/internal/
Dtfprof_show_multi.cc209 std::vector<string> input_types; in FormatInputShapes() local
/external/tensorflow/tensorflow/compiler/mlir/lite/transforms/
Dpost_quantize.cc92 llvm::SmallVector<Type, 4> input_types; in RemoveQuantizationAdaptorOps() local
/external/tensorflow/tensorflow/core/common_runtime/
Dlower_functional_ops_test.cc94 std::vector<DataType> input_types = {DT_INT32}; in WhileWithIfBody() local
Dgraph_view.cc193 uint8* input_types = item->input_type_base(); in InitializeNode() local
/external/mesa3d/src/compiler/nir/
Dnir_opcodes.py36 input_types, is_conversion, algebraic_properties, const_expr): argument
153 def opcode(name, output_size, output_type, input_sizes, input_types, argument
/external/tensorflow/tensorflow/compiler/mlir/lite/experimental/tac/transforms/
Draise_target_subgraphs.cc198 SmallVector<Type, 4> input_types; in BuildFuncOp() local
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/
Dtf_xla_mlir_translate.cc57 llvm::cl::opt<std::string> input_types( variable

123