| /external/tensorflow/tensorflow/contrib/lite/kernels/ |
| batch_to_space_nd_test.cc | 63 block_shape_ = AddConstInput(TensorType_INT32, block_shape, {2});
64 crops_ = AddConstInput(TensorType_INT32, crops, {2, 2});
86 block_shape_ = AddInput(TensorType_INT32);
87 crops_ = AddInput(TensorType_INT32);
|
| mean_test.cc | 55 axis_ = AddConstInput(TensorType_INT32, axis, axis_shape);
107 {TensorType_FLOAT32, {2}}, {TensorType_INT32, {4}},
122 {TensorType_FLOAT32, {3}}, {TensorType_INT32, {2}},
|
| space_to_batch_nd_test.cc | 63 block_shape_ = AddConstInput(TensorType_INT32, block_shape, {2});
64 paddings_ = AddConstInput(TensorType_INT32, paddings, {2, 2});
86 block_shape_ = AddInput(TensorType_INT32);
87 paddings_ = AddInput(TensorType_INT32);
|
| lsh_projection_test.cc | 36 input_ = AddInput(TensorType_INT32);
40 output_ = AddOutput(TensorType_INT32);
|
| embedding_lookup_sparse_test.cc | 39 lookup_ = AddInput(TensorType_INT32);
40 indices_ = AddInput(TensorType_INT32);
41 dense_shape_ = AddInput(TensorType_INT32);
|
| topk_v2_test.cc | 33 top_k_ = AddInput(TensorType_INT32);
34 output_indexes_ = AddOutput(TensorType_INT32);
133 TopKV2OpModel m({2, 3}, TensorType_INT32, 2);
|
| pad_test.cc | 57 paddings_ = AddConstInput(TensorType_INT32, paddings, paddings_shape);
78 paddings_ = AddInput(TensorType_INT32);
|
| resize_bilinear_test.cc | 33 size_ = AddConstInput(TensorType_INT32, size_data, {2});
35 size_ = AddInput({TensorType_INT32, {2}});
|
| split_test.cc | 33 axis_ = AddInput({TensorType_INT32, {1}});
35 axis_ = AddConstInput(TensorType_INT32, {axis}, {1});
|
| hashtable_lookup_test.cc | 39 lookup_ = AddInput(TensorType_INT32);
40 key_ = AddInput(TensorType_INT32);
|
| embedding_lookup_test.cc | 36 input_ = AddInput(TensorType_INT32);
|
| space_to_depth_test.cc | 76 SpaceToDepthOpModel m({TensorType_INT32, {1, 2, 2, 3}}, 2);
|
| depthwise_conv_test.cc | 45 TensorData bias{TensorType_INT32, {bias_size}, 0, 0, bias_scale};
|
| transpose_test.cc | 159 perm_ = AddConstInput(TensorType_INT32, perm, perm_shape);
179 perm_ = AddInput(TensorType_INT32);
|
| gather_test.cc | 32 positions_ = AddInput(TensorType_INT32);
|
| strided_slice_test.cc | 36 begin_ = AddInput(TensorType_INT32);
37 end_ = AddInput(TensorType_INT32);
38 strides_ = AddInput(TensorType_INT32);
|
| conv_test.cc | 61 TensorData bias{TensorType_INT32, {bias_size}, 0, 0, bias_scale};
|
| fully_connected_test.cc | 142 TensorData bias{TensorType_INT32, {units_}, 0, 0, bias_scale};
|
| test_util.cc | 66 } else if (t.type == TensorType_INT32) {
|
| /external/tensorflow/tensorflow/contrib/lite/models/smartreply/ops/ |
| predict_test.cc | 41 input_signature_ = AddInput(TensorType_INT32);
42 model_key_ = AddInput(TensorType_INT32);
43 model_label_ = AddInput(TensorType_INT32);
45 output_label_ = AddOutput(TensorType_INT32);
|
| extract_feature_test.cc | 39 signature_ = AddOutput(TensorType_INT32);
|
| /external/tensorflow/tensorflow/contrib/lite/toco/tflite/ |
| types.cc | 53 return ::tflite::TensorType_INT32;
71 case ::tflite::TensorType_INT32:
110 case ::tflite::TensorType_INT32:
|
| types_test.cc | 72 {ArrayDataType::kInt32, ::tflite::TensorType_INT32},
|
| /external/tensorflow/tensorflow/contrib/lite/tools/ |
| verifier.cc | 119 case TensorType_INT32:
|
| verifier_test.cc | 128 builder.AddTensor({2, 3}, TensorType_INT32, {}, "output");
191 CreateTensorDirect(builder, &shape, TensorType_INT32, /*buffer=*/2,
|
