Home
last modified time | relevance | path

Searched refs:input_types (Results 1 – 25 of 198) sorted by relevance

12345678

/external/tensorflow/tensorflow/core/framework/
Dfull_type_inference_util.cc44 return [i, n](const TypeRefVector& input_types, const TypeRefMap& type_vars) { in ReplicateInput() argument
45 const FullTypeDef& in_type = input_types.at(i).get(); in ReplicateInput()
58 return [](const TypeRefVector& input_types, in Merge()
60 DCHECK(!input_types.empty()); in Merge()
63 for (int i = 0; i < input_types.size(); i++) { in Merge()
64 const auto& t = input_types[i].get(); in Merge()
96 return [t, i](const TypeRefVector& input_types, in Encode()
98 DCHECK(input_types.size() >= i); in Encode()
101 const FullTypeDef& in_t = input_types[i].get(); in Encode()
117 return [t, i](const TypeRefVector& input_types, in Decode() argument
[all …]
Dnode_properties_test.cc63 const DataTypeVector& input_types, in ValidateNodeProperties() argument
67 ASSERT_EQ(props.input_types.size(), input_types.size()); in ValidateNodeProperties()
68 for (int i = 0; i < input_types.size(); ++i) { in ValidateNodeProperties()
69 EXPECT_EQ(props.input_types[i], input_types[i]); in ValidateNodeProperties()
70 EXPECT_EQ(props.input_types_slice[i], input_types[i]); in ValidateNodeProperties()
85 DataTypeVector input_types{DT_FLOAT, DT_INT32}; in TEST() local
87 DataTypeSlice input_types_slice(input_types); in TEST()
93 ValidateNodeProperties(props_from_slices, &op_def, node_def, input_types, in TEST()
97 NodeProperties props_from_vectors(&op_def, node_def, input_types, in TEST()
99 ValidateNodeProperties(props_from_vectors, &op_def, node_def, input_types, in TEST()
[all …]
Dnode_properties.cc29 DataTypeVector input_types; in CreateFromNodeDef() local
32 InOutTypesForNode(node_def, *op_def, &input_types, &output_types)); in CreateFromNodeDef()
34 std::move(input_types), in CreateFromNodeDef()
Dop_kernel_test_base.h75 DataTypeSlice input_types = {}) {
78 for (size_t i = 0; i < input_types.size(); ++i) {
92 EXPECT_EQ(input_types.size(), op->num_inputs());
147 DataTypeSlice input_types = {}) {
149 for (size_t i = 0; i < input_types.size(); ++i) {
Dnode_properties.h41 input_types(std::move(inputs)), in NodeProperties()
42 input_types_slice(input_types), in NodeProperties()
56 DataTypeVector input_types; member
/external/tensorflow/tensorflow/lite/toco/
Dtoco_cmdline_flags.cc95 Flag("input_types", parsed_flags.input_types.bind(), in ParseTocoFlagsFromCommandLineFlags()
96 parsed_flags.input_types.default_value(), in ParseTocoFlagsFromCommandLineFlags()
327 if (parsed_toco_flags.input_types.specified()) { in ReadTocoFlagsFromCommandLineFlags()
335 std::vector<std::string> input_types = in ReadTocoFlagsFromCommandLineFlags() local
336 absl::StrSplit(parsed_toco_flags.input_types.value(), ','); in ReadTocoFlagsFromCommandLineFlags()
337 QCHECK(!input_types.empty()); in ReadTocoFlagsFromCommandLineFlags()
338 for (size_t i = 1; i < input_types.size(); i++) { in ReadTocoFlagsFromCommandLineFlags()
339 QCHECK_EQ(input_types[i], input_types[0]); in ReadTocoFlagsFromCommandLineFlags()
342 QCHECK(toco::IODataType_Parse(input_types[0], &input_type)); in ReadTocoFlagsFromCommandLineFlags()
/external/tensorflow/tensorflow/lite/kernels/shim/test_op/
Dsimple_tflite_op_test.cc34 const std::vector<tflite::TensorType>& input_types, in SimpleOpModel() argument
41 for (const auto input_type : input_types) { in SimpleOpModel()
82 std::vector<tflite::TensorType> input_types = {tflite::TensorType_STRING, in TEST() local
92 SimpleOpModel m(/*op_options=*/builder.GetBuffer(), input_types, input_shapes, in TEST()
117 std::vector<tflite::TensorType> input_types = {tflite::TensorType_STRING}; in TEST() local
124 SimpleOpModel m(/*op_options=*/builder.GetBuffer(), input_types, input_shapes, in TEST()
/external/vixl/tools/test_generator/
Dparser.py54 def __init__(self, operand_types, operand_variants, input_types, argument
58 self.input_types = input_types
77 type_name = self.input_types[identifier]
151 input_types = {
160 return DataTypeBuilder(operand_types, operand_variants, input_types, input_values)
/external/tensorflow/tensorflow/compiler/mlir/tfr/passes/
Draise_to_tf.cc97 ArrayRef<Attribute> input_types,
164 const llvm::SmallVectorImpl<Attribute>& input_types, in CastValuesToSameType() argument
166 if (input_types.size() <= 1) return; in CastValuesToSameType()
168 Type target_input_type = input_types[0].cast<TypeAttr>().getValue(); in CastValuesToSameType()
170 for (auto i = 1; i < input_types.size(); ++i) { in CastValuesToSameType()
171 Type current_input_type = input_types[i].cast<TypeAttr>().getValue(); in CastValuesToSameType()
187 ArrayRef<Attribute> input_types, in AddDerivedAttrs() argument
196 derived_attrs->insert({attr_names[0].getValue(), input_types[0]}); in AddDerivedAttrs()
210 rewriter.getI32IntegerAttr(input_types.size())}); in AddDerivedAttrs()
213 derived_attrs->insert({attr_names[1].getValue(), input_types[0]}); in AddDerivedAttrs()
[all …]
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Delemental_ir_emitter.cc80 absl::Span<const PrimitiveType> input_types, PrimitiveType output_type, in EmitDeviceMathCall() argument
87 std::vector<PrimitiveType> converted_input_types(input_types.begin(), in EmitDeviceMathCall()
88 input_types.end()); in EmitDeviceMathCall()
93 if (input_types[i] == F16) { in EmitDeviceMathCall()
122 absl::Span<const PrimitiveType> input_types, PrimitiveType output_type) { in EmitLlvmIntrinsicMathCall() argument
140 return EmitMathCall(munged_callee, operands, input_types, output_type); in EmitLlvmIntrinsicMathCall()
145 absl::Span<const PrimitiveType> input_types, PrimitiveType output_type, in EmitMathCall() argument
148 for (PrimitiveType input_type : input_types) { in EmitMathCall()
157 callee_name, operands, input_types, output_type, in EmitMathCall()
Delemental_ir_emitter.h119 absl::Span<const PrimitiveType> input_types, PrimitiveType output_type);
126 absl::Span<const PrimitiveType> input_types, PrimitiveType output_type,
133 absl::Span<const PrimitiveType> input_types, PrimitiveType output_type,
/external/tensorflow/tensorflow/compiler/mlir/lite/utils/
Dperception_ops_utils_test.cc39 mlir::Builder* builder, const SmallVector<mlir::Type, NInput>& input_types, in createMaxUnpoolingFunc() argument
41 auto func_type = builder->getFunctionType(input_types, output_types); in createMaxUnpoolingFunc()
58 SmallVector<mlir::Type, 2> input_types{input_type, indices_type}; in createMaxUnpoolingFunc() local
60 return createMaxUnpoolingFunc<2, 1>(builder, input_types, output_types); in createMaxUnpoolingFunc()
146 SmallVector<mlir::Type, 1> input_types{input_type}; in TEST_F() local
150 createMaxUnpoolingFunc<1, 1>(builder_.get(), input_types, output_types); in TEST_F()
/external/tensorflow/tensorflow/core/common_runtime/
Dforward_type_inference.cc68 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
77 input_types.push_back(*no_type); in input_types()
90 input_types.at(in_edge->dst_input()) = t.args(in_edge->src_output()); in input_types()
94 return input_types; in input_types()
172 auto in_types = input_types(*n); in Run()
194 auto in_types = input_types(*n); in Run()
/external/tensorflow/tensorflow/python/framework/
Dfunction.py118 def __init__(self, *input_types, **kwargs): argument
148 self._input_types = input_types
259 input_types, argument
300 self._input_types = input_types
331 assert isinstance(input_types, (list, tuple))
332 self._arg_types = input_types
334 for i in range(len(input_types))]
657 def instantiate(self, input_types): argument
668 key = _type_list_to_str(input_types)
678 input_types,
[all …]
Dop_def_library_pybind.py26 attr_protos, inputs, input_types, output_structure = (
31 return attr_protos, inputs, input_types, output_structure
/external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/
DCompressElement.pbtxt5 type_list_attr: "input_types"
12 name: "input_types"
/external/tensorflow/tensorflow/python/data/ops/
Dstructured_function.py86 input_types=None, argument
126 if input_classes is None or input_shapes is None or input_types is None:
131 input_types, input_shapes, input_classes)
134 input_types is None):
141 input_shapes is None and input_types is None):
/external/tensorflow/tensorflow/core/kernels/data/experimental/
Dlist_dataset_op.cc46 const DataTypeVector& input_types, const DataTypeVector& output_types, in Dataset() argument
53 input_types_(input_types), in Dataset()
120 AttrValue input_types; in AsGraphDefInternal() local
121 b->BuildAttrValue(input_types_, &input_types); in AsGraphDefInternal()
123 {{kTinputTypes, input_types}}, output)); in AsGraphDefInternal()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dgroup_by_dialect.cc118 std::vector<Type> input_types; in wrapOpsInFunction() local
121 input_types.reserve(inputs.size()); in wrapOpsInFunction()
123 input_types.push_back(v.getType()); in wrapOpsInFunction()
137 builder.getFunctionType(input_types, output_types)); in wrapOpsInFunction()
/external/pytorch/aten/src/ATen/core/
Dclass_type.cpp59 std::stringstream input_types; in getSchemaInputTypesString() local
62 input_types << forward_args[i].type()->annotation_str(); in getSchemaInputTypesString()
64 input_types << ", "; in getSchemaInputTypesString()
68 input_types << "()"; in getSchemaInputTypesString()
70 return input_types.str(); in getSchemaInputTypesString()
76 std::string input_types = getSchemaInputTypesString(forward_schema); in getForwardPreHookErrorMessage() local
87 pre_hook_name + "(self, input: Tuple[" + input_types + "])"; in getForwardPreHookErrorMessage()
97 single_output + " or 'Tuple[" + input_types + "]'."; in getForwardPreHookErrorMessage()
104 std::string input_types = getSchemaInputTypesString(forward_schema); in getForwardHookErrorMessage() local
114 input_types + "], output: " + output_types + ")"; in getForwardHookErrorMessage()
/external/tensorflow/tensorflow/lite/toco/logging/
Dconversion_log_util.cc176 TFLITE_PROTO_NS::RepeatedPtrField<std::string>* input_types, in GetInputAndOutputTypes() argument
180 input_types->Add(ArrayDataTypeName(array.data_type)); in GetInputAndOutputTypes()
253 TFLITE_PROTO_NS::RepeatedPtrField<std::string> input_types, output_types; in PopulateConversionLog() local
254 GetInputAndOutputTypes(model, &input_types, &output_types); in PopulateConversionLog()
255 log->mutable_input_tensor_types()->CopyFrom(input_types); in PopulateConversionLog()
/external/tensorflow/tensorflow/python/ops/
Dwhile_v2.py1002 input_types=None, argument
1038 input_types=input_types,
1048 input_types=input_types,
1059 input_types=None, argument
1069 op_type, inputs, dtypes, input_types, name, attrs, op_def,
1089 input_types=input_types,
1105 input_types=None, argument
1116 if input_types is not None:
1117 input_types = tuple(input_types)
1128 return OptimizedReductionOpsCacheKey(op_type, inputs, dtypes, input_types,
/external/tensorflow/tensorflow/python/tpu/
Dtraining_loop.py64 input_types = [x.dtype for x in inputs]
143 if input_types != output_types:
146 "body: {} vs {}".format(input_types, output_types))
/external/tensorflow/tensorflow/compiler/mlir/lite/transforms/
Dpost_quantize.cc82 llvm::SmallVector<Type, 4> input_types; in RemoveQuantizationAdaptorOps() local
83 input_types.reserve(num_args); in RemoveQuantizationAdaptorOps()
99 input_types.push_back(quantize_type); in RemoveQuantizationAdaptorOps()
117 input_types.push_back(arg_type); in RemoveQuantizationAdaptorOps()
143 auto new_func_type = builder.getFunctionType(input_types, output_types); in RemoveQuantizationAdaptorOps()
/external/tensorflow/tensorflow/compiler/mlir/tfrt/jit/transforms/
Dtf_jitrt_legalize_i1_type.cc208 auto input_types = func_op.getFunctionType().getInputs(); in runOnOperation() local
211 input_types.begin(), input_types.end(), in runOnOperation()

12345678