Home
last modified time | relevance | path

Searched defs:input_types (Results 1 – 25 of 59) 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/compiler/mlir/lite/utils/
Dperception_ops_utils_test.cc37 mlir::Builder* builder, const SmallVector<mlir::Type, NInput>& input_types, in createMaxUnpoolingFunc()
57 SmallVector<mlir::Type, 2> input_types{input_type, indices_type}; in createMaxUnpoolingFunc() local
144 SmallVector<mlir::Type, 1> input_types{input_type}; in TEST_F() local
Dlstm_utils_test.cc59 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.cc81 absl::Span<const PrimitiveType> input_types, PrimitiveType output_type, in EmitDeviceMathCall()
123 absl::Span<const PrimitiveType> input_types, PrimitiveType output_type) { in EmitLlvmIntrinsicMathCall()
146 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/core/kernels/
Dremote_fused_graph_execute_op_test.cc41 DataTypeVector input_types({DT_FLOAT, DT_FLOAT}); in TEST_F() local
55 DataTypeVector input_types({DT_INT32, DT_INT32}); in TEST_F() local
112 DataTypeVector input_types{DT_FLOAT}; in BuildRemoteFusedGraphExecuteOp() local
/external/tensorflow/tensorflow/core/ops/
Dremote_fused_graph_ops_test.cc31 DataTypeVector input_types; in TEST() local
/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/framework/
Dfunction.py93 def __init__(self, *input_types, **kwargs):
232 input_types, argument
625 def instantiate(self, input_types): argument
810 input_types=None, argument
Dfunc_graph.py505 input_types=None, argument
538 input_types=None, argument
/external/tensorflow/tensorflow/python/ops/
Dwhile_v2.py953 input_types=None, argument
1010 input_types=None, argument
1056 input_types=None, argument
/external/tensorflow/tensorflow/lite/toco/
Dtoco_cmdline_flags.cc320 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
/external/tensorflow/tensorflow/compiler/mlir/tfr/passes/
Draise_to_tf.cc161 const llvm::SmallVectorImpl<Attribute>& input_types, in CastValuesToSameType()
184 ArrayRef<Attribute> input_types, in AddDerivedAttrs()
/external/tensorflow/tensorflow/core/grappler/optimizers/data/
Dmap_vectorization_test.cc519 std::vector<DataType> input_types({DT_INT64}); in TEST() local
538 std::vector<DataType> input_types({DT_INT64}); in TEST() local
557 std::vector<DataType> input_types({DT_INT64}); in TEST() local
/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.cc59 llvm::SmallVector<Type, 4> input_types; in RemoveQuantizationAdaptorOps() local
/external/tensorflow/tensorflow/core/kernels/hexagon/
Dgraph_transfer_utils.cc116 DataTypeVector input_types; in BuildFusedGraphDef() local
/external/tensorflow/tensorflow/core/common_runtime/
Dlower_functional_ops_test.cc94 std::vector<DataType> input_types = {DT_INT32}; in WhileWithIfBody() 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/lite/tools/versioning/
Dop_version.h27 std::vector<TensorType> input_types; member

123