Home
last modified time | relevance | path

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

12345

/external/mesa3d/prebuilt-intermediates/nir/
Dnir_opcodes.c179 .input_types = {
193 .input_types = {
207 .input_types = {
221 .input_types = {
235 .input_types = {
249 .input_types = {
263 .input_types = {
277 .input_types = {
291 .input_types = {
305 .input_types = {
[all …]
/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()
295 if (parsed_toco_flags.input_types.specified()) { in ReadTocoFlagsFromCommandLineFlags()
303 std::vector<string> input_types = in ReadTocoFlagsFromCommandLineFlags() local
304 absl::StrSplit(parsed_toco_flags.input_types.value(), ','); in ReadTocoFlagsFromCommandLineFlags()
305 QCHECK(!input_types.empty()); in ReadTocoFlagsFromCommandLineFlags()
306 for (int i = 1; i < input_types.size(); i++) { in ReadTocoFlagsFromCommandLineFlags()
307 QCHECK_EQ(input_types[i], input_types[0]); in ReadTocoFlagsFromCommandLineFlags()
310 QCHECK(toco::IODataType_Parse(input_types[0], &input_type)); in ReadTocoFlagsFromCommandLineFlags()
/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/core/kernels/data/
Ddataset_test_base.cc222 if (kernel.input_types().size() != inputs.size()) { in CheckOpKernelInput()
224 kernel.input_types().size(), in CheckOpKernelInput()
231 gtl::InlinedVector<TensorValue, 4>* inputs, DataTypeVector input_types, in AddDatasetInput() argument
233 if (input_types.size() < inputs->size()) { in AddDatasetInput()
235 inputs->size(), " vs. ", input_types.size()); in AddDatasetInput()
237 bool is_ref = IsRefType(input_types[inputs->size()]); in AddDatasetInput()
242 DataType expected_dtype = RemoveRefType(input_types[inputs->size()]); in AddDatasetInput()
249 if (input_types[inputs->size()] != dtype) { in AddDatasetInput()
252 " , but expected: ", input_types[inputs->size()]); in AddDatasetInput()
Ddataset_test_base.h96 &range_inputs, range_kernel->input_types(), TensorShape({}), {start})); in CreateRangeDataset()
98 &range_inputs, range_kernel->input_types(), TensorShape({}), {end})); in CreateRangeDataset()
100 &range_inputs, range_kernel->input_types(), TensorShape({}), {step})); in CreateRangeDataset()
164 DataTypeVector input_types, in AddDatasetInputFromArray() argument
168 AddDatasetInput(inputs, input_types, DataTypeToEnum<T>::v(), shape)); in AddDatasetInputFromArray()
177 DataTypeVector input_types, DataType dtype,
/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)
Dgenerator.py252 def GenerateInputs(self, input_types): argument
274 for input_type in input_types
277 args=", ".join(input_types.GetNames()),
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Delemental_ir_emitter.cc79 absl::Span<const PrimitiveType> input_types, PrimitiveType output_type) { in EmitLibdeviceMathCall() argument
88 std::vector<PrimitiveType> converted_input_types(input_types.begin(), in EmitLibdeviceMathCall()
89 input_types.end()); in EmitLibdeviceMathCall()
94 if (input_types[i] == F16) { in EmitLibdeviceMathCall()
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
147 for (PrimitiveType input_type : input_types) { in EmitMathCall()
156 callee_name, operands, input_types, output_type, in EmitMathCall()
286 absl::Span<const PrimitiveType> input_types, PrimitiveType output_type, in EmitDeviceFunctionCall() argument
[all …]
Delemental_ir_emitter.h117 absl::Span<const PrimitiveType> input_types, PrimitiveType output_type);
124 absl::Span<const PrimitiveType> input_types, PrimitiveType output_type);
130 absl::Span<const PrimitiveType> input_types, PrimitiveType output_type);
/external/tensorflow/tensorflow/python/framework/
Dfunction.py93 def __init__(self, *input_types, **kwargs): argument
123 self._input_types = input_types
208 input_types, argument
249 self._input_types = input_types
279 assert isinstance(input_types, (list, tuple))
280 self._arg_types = input_types
282 for i in range(len(input_types))]
583 def instantiate(self, input_types): argument
594 key = _type_list_to_str(input_types)
604 input_types,
[all …]
/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.cc90 const DataTypeVector& input_types) { in PrepOp() argument
116 CHECK_GT(input_types.size(), inputs->size()); in PrepOp()
117 CHECK_EQ(input_types[inputs->size()], dtype); in PrepOp()
120 CHECK_EQ(input_types[inputs->size()], dtype); in PrepOp()
/external/mesa3d/src/compiler/nir/
Dnir_opcodes.py35 input_types, algebraic_properties, const_expr): argument
70 assert isinstance(input_types, list)
71 assert isinstance(input_types[0], str)
74 assert len(input_sizes) == len(input_types)
85 self.input_types = input_types
107 def opcode(name, output_size, output_type, input_sizes, input_types, argument
111 input_types, algebraic_properties, const_expr)
Dnir_search_helpers.h49 switch (nir_op_infos[instr->op].input_types[src]) { in is_pos_power_of_two()
79 switch (nir_op_infos[instr->op].input_types[src]) { in is_neg_power_of_two()
104 switch (nir_op_infos[instr->op].input_types[src]) { in is_zero_to_one()
/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/lite/tools/accuracy/
Dutils_test.cc45 ASSERT_EQ(4, model_info.input_types.size()); in TEST()
49 DataType dataType = model_info.input_types[i]; in TEST()
/external/tensorflow/tensorflow/core/grappler/optimizers/data/
Dmap_vectorization.cc356 DataTypeVector input_types; in AddBranch() local
359 TF_RETURN_IF_ERROR(InputTypesForNode(*node, *op_def, &input_types)); in AddBranch()
360 DCHECK_EQ(input_types.size(), input_size); in AddBranch()
370 input_arg->set_type(input_types[i]); in AddBranch()
374 t_arguments->reserve(t_arguments->size() + input_types.size() - 1); in AddBranch()
375 t_arguments->insert(t_arguments->end(), input_types.begin() + 1, in AddBranch()
376 input_types.end()); in AddBranch()
Dmap_vectorization_test.cc431 std::vector<DataType> input_types({DT_INT64}); in TEST() local
432 auto input_node = AddArbitraryInputNode(&graph, nullptr, &input_types); in TEST()
451 std::vector<DataType> input_types({DT_INT64}); in TEST() local
452 auto input_node = AddArbitraryInputNode(&graph, &input_shapes, &input_types); in TEST()
471 std::vector<DataType> input_types({DT_INT64}); in TEST() local
472 auto input_node = AddArbitraryInputNode(&graph, &input_shapes, &input_types); in TEST()
/external/tensorflow/tensorflow/python/eager/
Dgraph_only_ops.py35 op = g.create_op("ZerosLike", [tensor], [dtype], input_types=[dtype],
50 op = g.create_op("Placeholder", [], [dtype], input_types=[],
/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/core/profiler/internal/
Dtfprof_show_multi.cc210 std::vector<string> input_types; in FormatInputShapes() local
211 input_types.reserve(shape_count_vec.size()); in FormatInputShapes()
214 input_types.push_back(strings::Printf( in FormatInputShapes()
218 return str_util::Join(input_types, "\n"); in FormatInputShapes()
/external/tensorflow/tensorflow/tools/graph_transforms/
Dinsert_logging.cc112 DataTypeVector input_types; in InsertLogging() local
114 TF_RETURN_IF_ERROR(GetInOutTypes(node, &input_types, &output_types)); in InsertLogging()
/external/tensorflow/tensorflow/compiler/jit/
Dxla_cluster_util.cc217 return std::find(node.input_types().begin(), node.input_types().end(), in HasResourceInputOrOutput()
218 DT_RESOURCE) != node.input_types().end() || in HasResourceInputOrOutput()
/external/tensorflow/tensorflow/contrib/copy_graph/python/util/
Dcopy_elements.py210 input_types = op._input_types[:]
218 new_control_inputs, input_types, new_original_op,
/external/tensorflow/tensorflow/lite/tools/accuracy/ilsvrc/
Dimagenet_model_evaluator.cc192 tfl_model_params.input_type = {model_info.input_types[0]}; in EvaluateModelForShard()
193 tfl_model_params.output_type = {model_info.input_types[0]}; in EvaluateModelForShard()
198 model_info.input_types[0]); in EvaluateModelForShard()

12345