| /external/mesa3d/prebuilt-intermediates/nir/ |
| nir_opcodes.c | 13 .input_types = {
27 .input_types = {
41 .input_types = {
55 .input_types = {
69 .input_types = {
83 .input_types = {
97 .input_types = {
111 .input_types = {
125 .input_types = {
139 .input_types =
[all...] |
| /external/tensorflow/tensorflow/core/ops/ |
| remote_fused_graph_ops_test.cc | 31 DataTypeVector input_types;
34 input_types.emplace_back(DT_FLOAT);
38 input_types.emplace_back(DT_INT32);
46 .Attr("Tinputs", input_types)
|
| /external/tensorflow/tensorflow/contrib/lite/toco/ |
| toco_cmdline_flags.cc | 74 Flag("input_types", parsed_flags.input_types.bind(),
75 parsed_flags.input_types.default_value(),
190 if (parsed_toco_flags.input_types.specified()) {
192 << "--input_types is deprecated. It was an ambiguous flag that set "
198 std::vector<string> input_types = local
199 absl::StrSplit(parsed_toco_flags.input_types.value(), ',');
200 QCHECK(!input_types.empty());
201 for (int i = 1; i < input_types.size(); i++) {
202 QCHECK_EQ(input_types[i], input_types[0])
[all...] |
| /external/mesa3d/src/compiler/nir/ |
| nir_search_helpers.h | 49 switch (nir_op_infos[instr->op].input_types[src]) {
79 switch (nir_op_infos[instr->op].input_types[src]) {
104 switch (nir_op_infos[instr->op].input_types[src]) {
|
| nir_opcodes.py | 36 input_types, algebraic_properties, const_expr):
41 - input_types is a list of types
71 assert isinstance(input_types, list)
72 assert isinstance(input_types[0], str)
75 assert len(input_sizes) == len(input_types)
86 self.input_types = input_types
107 def opcode(name, output_size, output_type, input_sizes, input_types,
111 input_types, algebraic_properties, const_expr)
|
| nir_algebraic.py | 348 src_type_bits = type_bits(nir_op.input_types[i])
391 src_type_bits = type_bits(nir_op.input_types[i])
421 src_type_bits = type_bits(nir_op.input_types[i])
460 src_type_bits = type_bits(nir_op.input_types[i])
|
| nir_lower_to_source_mods.c | 57 switch (nir_alu_type_get_base_type(nir_op_infos[alu->op].input_types[i])) {
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| build_xla_launch_ops_pass.cc | 76 DataTypeVector const_dtypes(node->input_types().begin(),
77 node->input_types().begin() + num_constant_args);
79 node->input_types().begin() + num_constant_args,
80 node->input_types().begin() + num_constant_args + num_nonconst_args);
|
| /external/tensorflow/tensorflow/python/eager/ |
| graph_only_ops.py | 35 op = g.create_op("ZerosLike", [tensor], [dtype], input_types=[dtype],
51 op = g.create_op("Placeholder", [], [dtype], input_types=[],
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| elemental_ir_emitter.h | 105 tensorflow::gtl::ArraySlice<PrimitiveType> input_types,
114 tensorflow::gtl::ArraySlice<PrimitiveType> input_types,
122 tensorflow::gtl::ArraySlice<PrimitiveType> input_types,
|
| elemental_ir_emitter.cc | 72 tensorflow::gtl::ArraySlice<PrimitiveType> input_types,
82 std::vector<PrimitiveType> converted_input_types(input_types.begin(),
83 input_types.end());
88 if (input_types[i] == F16) {
117 tensorflow::gtl::ArraySlice<PrimitiveType> input_types,
136 return EmitMathCall(munged_callee, operands, input_types, output_type);
142 tensorflow::gtl::ArraySlice<PrimitiveType> input_types,
145 for (PrimitiveType input_type : input_types) {
154 callee_name, operands, input_types, output_type,
274 tensorflow::gtl::ArraySlice<PrimitiveType> input_types,
[all...] |
| /external/tensorflow/tensorflow/contrib/tpu/python/tpu/ |
| training_loop.py | 59 input_types = [x.dtype for x in inputs]
142 if input_types != output_types:
145 "body: {} vs {}".format(input_types, output_types))
|
| tpu.py | 243 input_types = [x.dtype for x in inputs[0]]
244 input_arity = len(input_types)
252 if types != input_types:
256 input_types, i, types))
|
| /external/tensorflow/tensorflow/core/kernels/ |
| remote_fused_graph_execute_op_test.cc | 41 DataTypeVector input_types({DT_FLOAT, DT_FLOAT});
46 .Attr("Tinputs", input_types)
55 DataTypeVector input_types({DT_INT32, DT_INT32});
60 .Attr("Tinputs", input_types)
112 DataTypeVector input_types{DT_FLOAT};
117 .Attr("Tinputs", input_types)
|
| /external/vixl/tools/test_generator/ |
| parser.py | 49 input_types Dictionnary of type names corresponding to the JSON
54 def __init__(self, operand_types, operand_variants, input_types,
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)
|
| generator.py | 252 def GenerateInputs(self, input_types):
265 input_types The `InputList` object describing the list of inputs the
274 for input_type in input_types
277 args=", ".join(input_types.GetNames()),
|
| /external/tensorflow/tensorflow/python/framework/ |
| function.py | 91 def __init__(self, *input_types, **kwargs):
95 *input_types: A list of `tf.DType`
121 self._input_types = input_types
204 input_types,
217 input_types: The function's argument types. Can be a tuple, list of
239 self._input_types = input_types
260 assert isinstance(input_types, (list, tuple))
261 for i in range(len(input_types)):
263 argtype = input_types[i]
563 def instantiate(self, input_types)
[all...] |
| op_def_library.py | 393 input_types = []
642 input_types.extend(types)
644 input_types.extend(base_types)
786 input_types=input_types, attrs=attr_protos,
|
| /external/tensorflow/tensorflow/contrib/copy_graph/python/util/ |
| copy_elements.py | 210 input_types = op._input_types[:]
218 new_control_inputs, input_types, new_original_op,
|
| /external/tensorflow/tensorflow/tools/graph_transforms/ |
| insert_logging.cc | 112 DataTypeVector input_types; local
114 TF_RETURN_IF_ERROR(GetInOutTypes(node, &input_types, &output_types));
|
| /external/tensorflow/tensorflow/core/profiler/internal/ |
| tfprof_show_multi.cc | 210 std::vector<string> input_types; local
211 input_types.reserve(shape_count_vec.size());
214 input_types.push_back(strings::Printf(
218 return str_util::Join(input_types, "\n");
|
| /external/tensorflow/tensorflow/core/kernels/hexagon/ |
| graph_transfer_utils.cc | 114 DataTypeVector input_types; local
126 input_types.push_back(input_node_info.second.dtype());
151 .Attr("Tinputs", input_types)
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| direct_session.h | 140 DataTypeVector input_types; member in struct:tensorflow::DirectSession::ExecutorsAndKeys
216 RunStateArgs* run_state_args, DataTypeVector* input_types,
|
| /external/tensorflow/tensorflow/core/graph/ |
| graph_partition.cc | 107 MemoryTypeMap input_types; member in struct:tensorflow::__anon39621::GraphInfo
134 auto dst_it = info.input_types.find({dst->id(), dst_port});
135 DCHECK(dst_it != info.input_types.end());
148 auto dst_it = info.input_types.find({dst->id(), dst_port});
149 DCHECK(dst_it != info.input_types.end());
254 auto dst_it = g_info.input_types.find({dst->id(), dst_port});
255 DCHECK(dst_it != g_info.input_types.end());
553 info->input_types[{node_id, i}] = input_memory_types[i];
[all...] |
| graph.cc | 42 input_types(inputs.begin(), inputs.end()),
47 const DataTypeVector input_types; member in class:tensorflow::NodeProperties
145 int32 Node::num_inputs() const { return props_->input_types.size(); }
146 DataType Node::input_type(int32 i) const { return props_->input_types[i]; }
147 const DataTypeVector& Node::input_types() const { return props_->input_types; } function in class:tensorflow::Node
|
