| /external/tensorflow/tensorflow/core/framework/ |
| D | full_type_inference_util.cc | 44 return [i, n](const TypeRefVector& input_types, const TypeRefMap& type_vars) { in ReplicateInput()
59 const TypeRefMap& type_vars) -> StatusOr<FullTypeDef> { in Merge()
117 return [t, i](const TypeRefVector& input_types, in Decode()
118 const TypeRefMap& type_vars) -> StatusOr<FullTypeDef> { in Decode()
162 const TypeRefVector& input_types, in UnaryContainerAdd()
163 const TypeRefMap& type_vars) -> StatusOr<FullTypeDef> { in UnaryContainerAdd()
231 const TypeRefMap& type_vars) -> StatusOr<FullTypeDef> { in MultiaryUnstack()
264 const TypeRefMap& type_vars) -> StatusOr<FullTypeDef> { in ContainerMap()
300 const TypeRefMap& type_vars) -> StatusOr<FullTypeDef> { in MapCovariant()
|
| D | node_properties_test.cc | 63 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
|
| D | node_properties.cc | 29 DataTypeVector input_types; in CreateFromNodeDef() local
|
| D | node_properties.h | 56 DataTypeVector input_types; member
|
| /external/tensorflow/tensorflow/lite/kernels/shim/test_op/ |
| D | simple_tflite_op_test.cc | 34 const std::vector<tflite::TensorType>& input_types, in SimpleOpModel()
82 std::vector<tflite::TensorType> input_types = {tflite::TensorType_STRING, in TEST() local
117 std::vector<tflite::TensorType> input_types = {tflite::TensorType_STRING}; in TEST() local
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/utils/ |
| D | perception_ops_utils_test.cc | 39 mlir::Builder* builder, const SmallVector<mlir::Type, NInput>& input_types, in createMaxUnpoolingFunc()
58 SmallVector<mlir::Type, 2> input_types{input_type, indices_type}; in createMaxUnpoolingFunc() local
146 SmallVector<mlir::Type, 1> input_types{input_type}; in TEST_F() local
|
| D | lstm_utils_test.cc | 62 SmallVector<mlir::Type, 4> input_types{input_type, weight_type, bias_type, in createLstmCompositeFunc() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | elemental_ir_emitter.cc | 80 absl::Span<const PrimitiveType> input_types, PrimitiveType output_type, in EmitDeviceMathCall()
122 absl::Span<const PrimitiveType> input_types, PrimitiveType output_type) { in EmitLlvmIntrinsicMathCall()
145 absl::Span<const PrimitiveType> input_types, PrimitiveType output_type, in EmitMathCall()
|
| D | target_util.cc | 196 absl::Span<const PrimitiveType> input_types, PrimitiveType output_type, in EmitDeviceFunctionCall()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | forward_type_inference.cc | 68 std::vector<std::reference_wrapper<const FullTypeDef>> input_types( in input_types() function
72 std::vector<std::reference_wrapper<const FullTypeDef>> input_types; in input_types() local
|
| D | forward_type_inference_test.cc | 185 .SetForwardTypeFn([](const TypeRefVector& input_types,
197 .SetForwardTypeFn([](const TypeRefVector& input_types,
198 const TypeRefMap& type_vars) { in __anonc63ccbf50402()
|
| /external/tensorflow/tensorflow/python/framework/ |
| D | function.py | 118 def __init__(self, *input_types, **kwargs):
259 input_types, argument
657 def instantiate(self, input_types): argument
842 input_types=None, argument
|
| /external/tensorflow/tensorflow/lite/toco/logging/ |
| D | conversion_log_util.cc | 176 TFLITE_PROTO_NS::RepeatedPtrField<std::string>* input_types, in GetInputAndOutputTypes()
253 TFLITE_PROTO_NS::RepeatedPtrField<std::string> input_types, output_types; in PopulateConversionLog() local
|
| /external/tensorflow/tensorflow/core/kernels/data/experimental/ |
| D | list_dataset_op.cc | 46 const DataTypeVector& input_types, const DataTypeVector& output_types, in Dataset()
120 AttrValue input_types; in AsGraphDefInternal() local
|
| /external/pytorch/test/mobile/nnc/ |
| D | test_nnc_backend.cpp | 26 const std::string& input_types, in create_compile_spec()
|
| /external/pytorch/aten/src/ATen/core/ |
| D | class_type.cpp | 59 std::stringstream input_types; in getSchemaInputTypesString() local
76 std::string input_types = getSchemaInputTypesString(forward_schema); in getForwardPreHookErrorMessage() local
104 std::string input_types = getSchemaInputTypesString(forward_schema); in getForwardHookErrorMessage() local
|
| /external/tensorflow/tensorflow/lite/toco/ |
| D | toco_cmdline_flags.cc | 335 std::vector<std::string> input_types = in ReadTocoFlagsFromCommandLineFlags() local
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | while_v2.py | 1002 input_types=None, argument
1059 input_types=None, argument
1105 input_types=None, argument
|
| /external/tensorflow/tensorflow/tools/graph_transforms/ |
| D | insert_logging.cc | 111 DataTypeVector input_types; in InsertLogging() local
|
| /external/vixl/tools/test_generator/ |
| D | parser.py | 54 def __init__(self, operand_types, operand_variants, input_types, argument
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | group_by_dialect.cc | 118 std::vector<Type> input_types; in wrapOpsInFunction() local
|
| /external/tensorflow/tensorflow/python/data/ops/ |
| D | structured_function.py | 86 input_types=None, argument
|
| /external/tensorflow/tensorflow/compiler/mlir/tfr/passes/ |
| D | raise_to_tf.cc | 164 const llvm::SmallVectorImpl<Attribute>& input_types, in CastValuesToSameType()
187 ArrayRef<Attribute> input_types, in AddDerivedAttrs()
|
| /external/tensorflow/tensorflow/core/profiler/internal/ |
| D | tfprof_show_multi.cc | 209 std::vector<string> input_types; in FormatInputShapes() local
|
| /external/rust/android-crates-io/crates/googletest_macro/src/ |
| D | lib.rs | 290 let input_types = signature in closure_body() localVariable
|