Home
last modified time | relevance | path

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

123456

/external/mesa3d/prebuilt-intermediates/nir/
Dnir_opcodes.c303 .input_types = {
318 .input_types = {
333 .input_types = {
348 .input_types = {
363 .input_types = {
378 .input_types = {
393 .input_types = {
408 .input_types = {
423 .input_types = {
438 .input_types = {
[all …]
/external/tensorflow/tensorflow/lite/tools/versioning/
Dop_version.cc34 if (op_sig.input_types.at(0) == TensorType_INT8 && in GetBuiltinOperatorVersion()
35 op_sig.input_types.at(1) == TensorType_INT8 && in GetBuiltinOperatorVersion()
41 if (op_sig.input_types.at(0) == TensorType_FLOAT32 && in GetBuiltinOperatorVersion()
42 op_sig.input_types.at(1) == TensorType_INT8 && in GetBuiltinOperatorVersion()
51 if (op_sig.input_types.at(0) == TensorType_FLOAT32 && in GetBuiltinOperatorVersion()
52 op_sig.input_types.at(1) == TensorType_INT8 && in GetBuiltinOperatorVersion()
58 if (op_sig.input_types.at(0) == TensorType_INT8 && in GetBuiltinOperatorVersion()
59 op_sig.input_types.at(1) == TensorType_INT8 && in GetBuiltinOperatorVersion()
86 if (op_sig.input_types.size() == 2) { in GetBuiltinOperatorVersion()
94 if (op_sig.input_types.at(0) == TensorType_INT8 && in GetBuiltinOperatorVersion()
[all …]
Dop_version_test.cc27 .input_types = std::vector<TensorType>{TensorType_INT8, TensorType_INT8, in TEST()
34 .input_types = std::vector<TensorType>{TensorType_UINT8, TensorType_UINT8, in TEST()
41 .input_types = std::vector<TensorType>{TensorType_INT64, TensorType_INT64, in TEST()
48 .input_types = std::vector<TensorType>{TensorType_INT32, TensorType_INT32, in TEST()
59 .input_types = std::vector<TensorType>{TensorType_INT8}, in SimpleVersioningTest()
65 .input_types = std::vector<TensorType>{TensorType_UINT8}, in SimpleVersioningTest()
74 .input_types = std::vector<TensorType>{}, in SimpleOutputVersioningTest()
81 .input_types = std::vector<TensorType>{}, in SimpleOutputVersioningTest()
126 .input_types = std::vector<TensorType>{TensorType_INT8}, in TEST()
132 .input_types = std::vector<TensorType>{TensorType_UINT8}, in TEST()
[all …]
/external/v8/src/torque/
Dcfg.h27 base::Optional<Stack<const Type*>> input_types, in Block() argument
30 input_types_(std::move(input_types)), in Block()
40 void SetInputTypes(const Stack<const Type*>& input_types);
98 explicit ControlFlowGraph(Stack<const Type*> input_types) { in ControlFlowGraph() argument
99 start_ = NewBlock(std::move(input_types), false); in ControlFlowGraph()
103 Block* NewBlock(base::Optional<Stack<const Type*>> input_types, in NewBlock() argument
105 blocks_.emplace_back(this, next_block_id_++, std::move(input_types), in NewBlock()
145 explicit CfgAssembler(Stack<const Type*> input_types) in CfgAssembler() argument
146 : current_stack_(std::move(input_types)), cfg_(current_stack_) {} in CfgAssembler()
159 base::Optional<Stack<const Type*>> input_types = base::nullopt,
[all …]
Dcfg.cc13 void Block::SetInputTypes(const Stack<const Type*>& input_types) { in SetInputTypes() argument
15 input_types_ = input_types; in SetInputTypes()
17 } else if (*input_types_ == input_types) { in SetInputTypes()
21 DCHECK_EQ(input_types.Size(), input_types_->Size()); in SetInputTypes()
24 auto c2_iterator = input_types.begin(); in SetInputTypes()
42 for (intptr_t i = std::max(input_types_->Size(), input_types.Size()) - 1; in SetInputTypes()
46 if (static_cast<size_t>(i) < input_types.Size()) { in SetInputTypes()
47 left = input_types.Peek(BottomOffset{static_cast<size_t>(i)}); in SetInputTypes()
/external/tensorflow/tensorflow/core/ops/
Dremote_fused_graph_ops_test.cc31 DataTypeVector input_types; in TEST() local
34 input_types.emplace_back(DT_FLOAT); in TEST()
38 input_types.emplace_back(DT_INT32); in TEST()
46 .Attr("Tinputs", input_types) in TEST()
/external/tensorflow/tensorflow/lite/toco/
Dtoco_cmdline_flags.cc93 Flag("input_types", parsed_flags.input_types.bind(), in ParseTocoFlagsFromCommandLineFlags()
94 parsed_flags.input_types.default_value(), in ParseTocoFlagsFromCommandLineFlags()
312 if (parsed_toco_flags.input_types.specified()) { in ReadTocoFlagsFromCommandLineFlags()
320 std::vector<string> input_types = in ReadTocoFlagsFromCommandLineFlags() local
321 absl::StrSplit(parsed_toco_flags.input_types.value(), ','); in ReadTocoFlagsFromCommandLineFlags()
322 QCHECK(!input_types.empty()); in ReadTocoFlagsFromCommandLineFlags()
323 for (int i = 1; i < input_types.size(); i++) { in ReadTocoFlagsFromCommandLineFlags()
324 QCHECK_EQ(input_types[i], input_types[0]); in ReadTocoFlagsFromCommandLineFlags()
327 QCHECK(toco::IODataType_Parse(input_types[0], &input_type)); in ReadTocoFlagsFromCommandLineFlags()
/external/tensorflow/tensorflow/compiler/mlir/lite/transforms/
Dpost_quantize.cc58 llvm::SmallVector<Type, 4> input_types; in RemoveQuantizationAdaptorOps() local
59 input_types.reserve(num_args); in RemoveQuantizationAdaptorOps()
75 input_types.push_back(quantize_type); in RemoveQuantizationAdaptorOps()
93 input_types.push_back(arg_type); in RemoveQuantizationAdaptorOps()
117 auto new_func_type = builder.getFunctionType(input_types, output_types); in RemoveQuantizationAdaptorOps()
/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/core/kernels/
Dremote_fused_graph_execute_op_test.cc41 DataTypeVector input_types({DT_FLOAT, DT_FLOAT}); in TEST_F() local
46 .Attr("Tinputs", input_types) in TEST_F()
55 DataTypeVector input_types({DT_INT32, DT_INT32}); in TEST_F() local
60 .Attr("Tinputs", input_types) in TEST_F()
112 DataTypeVector input_types{DT_FLOAT}; in BuildRemoteFusedGraphExecuteOp() local
117 .Attr("Tinputs", input_types) in BuildRemoteFusedGraphExecuteOp()
Dscoped_allocator_ops_test.cc91 const DataTypeVector& input_types) { in PrepOp() argument
117 CHECK_GT(input_types.size(), inputs->size()); in PrepOp()
118 CHECK_EQ(input_types[inputs->size()], dtype); in PrepOp()
121 CHECK_EQ(input_types[inputs->size()], dtype); in PrepOp()
/external/tensorflow/tensorflow/lite/toco/logging/
Dconversion_log_util.cc171 const Model& model, TFLITE_PROTO_NS::RepeatedPtrField<string>* input_types, in GetInputAndOutputTypes() argument
175 input_types->Add(ArrayDataTypeName(array.data_type)); in GetInputAndOutputTypes()
226 TFLITE_PROTO_NS::RepeatedPtrField<string> input_types, output_types; in PopulateConversionLog() local
227 GetInputAndOutputTypes(model, &input_types, &output_types); in PopulateConversionLog()
228 log->mutable_input_tensor_types()->CopyFrom(input_types); in PopulateConversionLog()
/external/tensorflow/tensorflow/python/framework/
Dfunction.py93 def __init__(self, *input_types, **kwargs): argument
123 self._input_types = input_types
230 input_types, argument
271 self._input_types = input_types
302 assert isinstance(input_types, (list, tuple))
303 self._arg_types = input_types
305 for i in range(len(input_types))]
627 def instantiate(self, input_types): argument
638 key = _type_list_to_str(input_types)
648 input_types,
[all …]
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Delemental_ir_emitter.cc80 absl::Span<const PrimitiveType> input_types, PrimitiveType output_type) { in EmitDeviceMathCall() argument
86 std::vector<PrimitiveType> converted_input_types(input_types.begin(), in EmitDeviceMathCall()
87 input_types.end()); in EmitDeviceMathCall()
92 if (input_types[i] == F16) { in EmitDeviceMathCall()
121 absl::Span<const PrimitiveType> input_types, PrimitiveType output_type) { in EmitLlvmIntrinsicMathCall() argument
139 return EmitMathCall(munged_callee, operands, input_types, output_type); in EmitLlvmIntrinsicMathCall()
144 absl::Span<const PrimitiveType> input_types, PrimitiveType output_type) { in EmitMathCall() argument
146 for (PrimitiveType input_type : input_types) { in EmitMathCall()
155 callee_name, operands, input_types, output_type, in EmitMathCall()
Delemental_ir_emitter.h108 absl::Span<const PrimitiveType> input_types, PrimitiveType output_type);
115 absl::Span<const PrimitiveType> input_types, PrimitiveType output_type);
121 absl::Span<const PrimitiveType> input_types, PrimitiveType output_type);
/external/tensorflow/tensorflow/python/tpu/
Dtraining_loop.py60 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/tensorflow/transforms/
Doptimize_global_tensors.cc140 auto input_types = old_ftype.getInputs().vec(); in FreezeGlobalTensors() local
141 input_types[arg_index] = new_type; in FreezeGlobalTensors()
143 builder.getFunctionType(input_types, old_ftype.getResults())); in FreezeGlobalTensors()
Dshape_inference.cc383 llvm::ArrayRef<Type> input_types, in RefineShapeForControlFlowFunc() argument
397 if (input_types == func_type.getInputs()) return success(); in RefineShapeForControlFlowFunc()
399 func.setType(FunctionType::get(input_types, func_type.getResults(), in RefineShapeForControlFlowFunc()
403 arg_and_idx.value().setType(input_types[arg_and_idx.index()]); in RefineShapeForControlFlowFunc()
412 func.setType(FunctionType::get(input_types, new_return_types.getValue(), in RefineShapeForControlFlowFunc()
420 ModuleOp module, Operation::operand_type_range input_types, in PropagateShapeToFunctions() argument
424 auto types = llvm::to_vector<4>(input_types); in PropagateShapeToFunctions()
/external/tensorflow/tensorflow/lite/experimental/delegates/hexagon/
Dutils.cc39 const std::vector<TfLiteType>& input_types) { in InputsWithCorrectTypes() argument
40 if (node->inputs->size != input_types.size()) return false; in InputsWithCorrectTypes()
41 for (int i = 0; i < input_types.size(); ++i) { in InputsWithCorrectTypes()
42 if (!TensorTypeMatch(node->inputs->data[i], context, input_types[i])) in InputsWithCorrectTypes()
/external/mesa3d/src/compiler/nir/
Dnir_opcodes.py36 input_types, is_conversion, algebraic_properties, const_expr): argument
72 assert isinstance(input_types, list)
73 assert isinstance(input_types[0], str)
77 assert len(input_sizes) == len(input_types)
88 self.input_types = input_types
153 def opcode(name, output_size, output_type, input_sizes, input_types, argument
157 input_types, is_conversion, algebraic_properties,
Dnir_search_helpers.h45 nir_alu_type type = nir_op_infos[instr->op].input_types[src]; in is_pos_power_of_two()
77 nir_alu_type type = nir_op_infos[instr->op].input_types[src]; in is_neg_power_of_two()
129 switch (nir_op_infos[instr->op].input_types[src]) { in is_zero_to_one()
160 switch (nir_op_infos[instr->op].input_types[src]) { in is_gt_0_and_lt_1()
184 nir_alu_type type = nir_op_infos[instr->op].input_types[src]; in is_not_const_zero()
/external/tensorflow/tensorflow/core/grappler/optimizers/data/
Dmap_vectorization.cc357 DataTypeVector input_types; in AddBranch() local
360 TF_RETURN_IF_ERROR(InputTypesForNode(*node, *op_def, &input_types)); in AddBranch()
361 DCHECK_EQ(input_types.size(), input_size); in AddBranch()
371 input_arg->set_type(input_types[i]); in AddBranch()
375 t_arguments->reserve(t_arguments->size() + input_types.size() - 1); in AddBranch()
376 t_arguments->insert(t_arguments->end(), input_types.begin() + 1, in AddBranch()
377 input_types.end()); in AddBranch()
/external/tensorflow/tensorflow/core/kernels/hexagon/
Dgraph_transfer_utils.cc116 DataTypeVector input_types; in BuildFusedGraphDef() local
128 input_types.push_back(input_node_info.second.dtype()); in BuildFusedGraphDef()
153 .Attr("Tinputs", input_types) in BuildFusedGraphDef()
/external/tensorflow/tensorflow/compiler/tf2tensorrt/plugin/
Dplugin_cast.cu.cc80 DataType getOutputDataType(int index, const DataType* input_types, in getOutputDataType() argument
99 const DataType* input_types, in configurePlugin() argument
106 DCHECK(src_type_ == input_types[0]); in configurePlugin()
/external/tensorflow/tensorflow/core/profiler/internal/
Dtfprof_show_multi.cc209 std::vector<string> input_types; in FormatInputShapes() local
210 input_types.reserve(shape_count_vec.size()); in FormatInputShapes()
213 input_types.push_back(absl::StrFormat( in FormatInputShapes()
217 return absl::StrJoin(input_types, "\n"); in FormatInputShapes()

123456