| /external/tensorflow/tensorflow/lite/kernels/ |
| D | pack.cc | 34 TF_LITE_ENSURE_EQ(context, NumInputs(node), data->values_count); in Prepare()
49 for (int i = 1; i < data->values_count; ++i) { in Prepare()
62 output_shape->data[index] = data->values_count; in Prepare()
73 for (int i = 0; i < data->values_count; i++) { in Prepare()
85 int values_count, int axis) { in PackImpl() argument
89 op_params.inputs_count = values_count; in PackImpl()
102 PackImpl<float>(context, node, output, data->values_count, data->axis); in Eval()
106 PackImpl<uint8_t>(context, node, output, data->values_count, data->axis); in Eval()
110 PackImpl<int8_t>(context, node, output, data->values_count, data->axis); in Eval()
114 PackImpl<int32_t>(context, node, output, data->values_count, data->axis); in Eval()
[all …]
|
| D | pack_test.cc | 30 PackOpModel(const TensorData& input_template, int axis, int values_count) { in PackOpModel() argument
32 for (int i = 0; i < values_count; ++i) { in PackOpModel()
39 CreatePackOptions(builder_, values_count, axis).Union()); in PackOpModel()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | quantization_utils_test.cc | 38 const int values_count = values_quantized.size(); in TestRequantizeMany() local
40 expected_values.reserve(values_count); in TestRequantizeMany()
41 for (int value_index = 0; value_index < values_count; ++value_index) { in TestRequantizeMany()
49 Tensor o_tensor(DT_QUINT8, TensorShape{values_count}); in TestRequantizeMany()
64 for (size_t value_index = 0; value_index < values_count; ++value_index) { in TestRequantizeMany()
80 const int values_count = values_quantized.size(); in TestRequantizeMany8To32Bit() local
82 expected_values.reserve(values_count); in TestRequantizeMany8To32Bit()
83 for (int value_index = 0; value_index < values_count; ++value_index) { in TestRequantizeMany8To32Bit()
91 Tensor o_tensor(DT_QINT32, TensorShape{values_count}); in TestRequantizeMany8To32Bit()
100 for (int value_index = 0; value_index < values_count; ++value_index) { in TestRequantizeMany8To32Bit()
[all …]
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | runtime_key_value_sort.cc | 34 int64 a, int64 b, int64 c, char** values, int32 values_count, in __xla_cpu_runtime_KeyValueSort() argument
40 TF_ANNOTATE_MEMORY_IS_INITIALIZED(values, values_count * sizeof(char*)); in __xla_cpu_runtime_KeyValueSort()
42 values_count * sizeof(int32)); in __xla_cpu_runtime_KeyValueSort()
58 std::unique_ptr<char*[]> comparison_values(new char*[2 * values_count]); in __xla_cpu_runtime_KeyValueSort()
77 for (int32 i = 0; i < values_count; ++i) { in __xla_cpu_runtime_KeyValueSort()
95 for (int32 idx = 0; idx < values_count; ++idx) { in __xla_cpu_runtime_KeyValueSort()
|
| D | runtime_key_value_sort.h | 40 char** values, tensorflow::int32 values_count,
|
| D | ir_emitter.cc | 532 sort->values_count() > 0 ? ShapeIndex({i}) : ShapeIndex({}); in HandleSort()
618 if (sort->values_count() > 0) { in HandleSort()
|
| /external/tensorflow/tensorflow/lite/toco/graph_transformations/ |
| D | unroll_batch_matmul.cc | 155 pack_op->values_count = pack_inputs.size(); in UnrollBatchMatMulRecursion()
289 pack_op->values_count = pack_inputs.size(); in Run()
|
| /external/tensorflow/tensorflow/lite/c/ |
| D | builtin_op_data.h | 350 int values_count; member
|
| /external/tensorflow/tensorflow/lite/core/api/ |
| D | flatbuffer_conversions.cc | 602 params->values_count = pack_params->values_count(); in ParseOpData()
|
| /external/tensorflow/tensorflow/lite/toco/tflite/ |
| D | operator_test.cc | 534 op.values_count = 3; in TEST_F()
538 EXPECT_EQ(op.values_count, output_toco_op->values_count); in TEST_F()
|
| D | operator.cc | 1664 return ::tflite::CreatePackOptions(*builder, op.values_count, op.axis); in WriteOptions()
1669 op->values_count = options.values_count(); in ReadOptions()
|
| /external/python/cpython2/Python/ |
| D | ast.c | 2280 int i, j, values_count, start = 1; in ast_for_print_stmt() local
2289 values_count = (NCH(n) + 1 - start) / 2; in ast_for_print_stmt()
2290 if (values_count) { in ast_for_print_stmt()
2291 seq = asdl_seq_new(values_count, c->c_arena); in ast_for_print_stmt()
|
| /external/icu/icu4c/source/test/intltest/ |
| D | tmsgfmt.cpp | 1529 int32_t values_count = UPRV_LENGTHOF(values); in TestRBNF() local
1553 for (int j = 0; j < values_count; ++j) { in TestRBNF()
|
| /external/tensorflow/tensorflow/lite/toco/ |
| D | model.h | 1328 int values_count;
|
| D | import_tensorflow.cc | 1958 op->values_count = HasAttr(node, "N") ? GetIntAttr(node, "N") : num_inputs; in ConvertPackOperator()
|
| /external/tensorflow/tensorflow/lite/schema/ |
| D | schema.fbs | 677 values_count:int;
|
| D | schema_generated.h | 6560 int32_t values_count;
6563 : values_count(0),
6574 int32_t values_count() const {
6594 void add_values_count(int32_t values_count) {
6595 fbb_.AddElement<int32_t>(PackOptions::VT_VALUES_COUNT, values_count, 0);
6614 int32_t values_count = 0,
6618 builder_.add_values_count(values_count);
10632 { auto _e = values_count(); _o->values_count = _e; };
10644 auto _values_count = _o->values_count;
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_instructions.h | 509 int64 values_count() const { return operand_count() - 1; } in values_count() function
|