/external/tensorflow/tensorflow/core/kernels/ |
D | quantize_and_dequantize_op_test.cc | 44 .Input(FakeInput(DT_FLOAT)) in TEST_F() 45 .Input(FakeInput(DT_FLOAT)) in TEST_F() 46 .Input(FakeInput(DT_FLOAT)) in TEST_F() 69 .Input(FakeInput(DT_FLOAT)) in TEST_F() 70 .Input(FakeInput(DT_FLOAT)) in TEST_F() 71 .Input(FakeInput(DT_FLOAT)) in TEST_F() 72 .Input(FakeInput(DT_INT32)) in TEST_F() 96 .Input(FakeInput(DT_FLOAT)) in TEST_F() 97 .Input(FakeInput(DT_FLOAT)) in TEST_F() 98 .Input(FakeInput(DT_FLOAT)) in TEST_F() [all …]
|
D | quantized_matmul_op_test.cc | 46 .Input(FakeInput(DT_QUINT8)) in TEST_F() 47 .Input(FakeInput(DT_QUINT8)) in TEST_F() 48 .Input(FakeInput(DT_FLOAT)) in TEST_F() 49 .Input(FakeInput(DT_FLOAT)) in TEST_F() 50 .Input(FakeInput(DT_FLOAT)) in TEST_F() 51 .Input(FakeInput(DT_FLOAT)) in TEST_F() 97 .Input(FakeInput(DT_QUINT8)) in TEST_F() 98 .Input(FakeInput(DT_QUINT8)) in TEST_F() 99 .Input(FakeInput(DT_FLOAT)) in TEST_F() 100 .Input(FakeInput(DT_FLOAT)) in TEST_F() [all …]
|
D | fake_quant_ops_test.cc | 58 .Input(FakeInput(DT_FLOAT)) // inputs in RunTestFakeQuantWithMinMaxArgs() 83 .Input(FakeInput(DT_FLOAT)) // inputs in RunTestFakeQuantWithMinMaxVars() 84 .Input(FakeInput(DT_FLOAT)) // min in RunTestFakeQuantWithMinMaxVars() 85 .Input(FakeInput(DT_FLOAT)) // max in RunTestFakeQuantWithMinMaxVars() 113 .Input(FakeInput(DT_FLOAT)) // inputs in RunTestFakeQuantWithMinMaxVarsPerChannel() 114 .Input(FakeInput(DT_FLOAT)) // min in RunTestFakeQuantWithMinMaxVarsPerChannel() 115 .Input(FakeInput(DT_FLOAT)) // max in RunTestFakeQuantWithMinMaxVarsPerChannel() 362 .Input(FakeInput(DT_FLOAT)) // gradient in TEST_F() 363 .Input(FakeInput(DT_FLOAT)) // inputs in TEST_F() 392 .Input(FakeInput(DT_FLOAT)) // gradient in TEST_F() [all …]
|
D | quantized_batch_norm_op_test.cc | 40 .Input(FakeInput(DT_QUINT8)) in TEST_F() 41 .Input(FakeInput(DT_FLOAT)) in TEST_F() 42 .Input(FakeInput(DT_FLOAT)) in TEST_F() 43 .Input(FakeInput(DT_QUINT8)) in TEST_F() 44 .Input(FakeInput(DT_FLOAT)) in TEST_F() 45 .Input(FakeInput(DT_FLOAT)) in TEST_F() 46 .Input(FakeInput(DT_QUINT8)) in TEST_F() 47 .Input(FakeInput(DT_FLOAT)) in TEST_F() 48 .Input(FakeInput(DT_FLOAT)) in TEST_F() 49 .Input(FakeInput(DT_QUINT8)) in TEST_F() [all …]
|
D | quantized_conv_ops_test.cc | 45 .Input(FakeInput(DT_QUINT8)) in TEST_F() 46 .Input(FakeInput(DT_QUINT8)) in TEST_F() 47 .Input(FakeInput(DT_FLOAT)) in TEST_F() 48 .Input(FakeInput(DT_FLOAT)) in TEST_F() 49 .Input(FakeInput(DT_FLOAT)) in TEST_F() 50 .Input(FakeInput(DT_FLOAT)) in TEST_F() 137 .Input(FakeInput(DT_QUINT8)) in TEST_F() 138 .Input(FakeInput(DT_QUINT8)) in TEST_F() 139 .Input(FakeInput(DT_FLOAT)) in TEST_F() 140 .Input(FakeInput(DT_FLOAT)) in TEST_F() [all …]
|
D | quantize_op_test.cc | 33 .Input(FakeInput(DT_FLOAT)) in TEST_F() 34 .Input(FakeInput(DT_FLOAT)) in TEST_F() 35 .Input(FakeInput(DT_FLOAT)) in TEST_F() 56 .Input(FakeInput(DT_FLOAT)) in TEST_F() 57 .Input(FakeInput(DT_FLOAT)) in TEST_F() 58 .Input(FakeInput(DT_FLOAT)) in TEST_F() 88 .Input(FakeInput(DT_FLOAT)) in TEST_F() 89 .Input(FakeInput(DT_FLOAT)) in TEST_F() 90 .Input(FakeInput(DT_FLOAT)) in TEST_F() 118 .Input(FakeInput(DT_FLOAT)) in TEST_F() [all …]
|
D | fused_batch_norm_op_test.cc | 34 .Input(FakeInput(DT_FLOAT)) in TEST_F() 35 .Input(FakeInput(DT_FLOAT)) in TEST_F() 36 .Input(FakeInput(DT_FLOAT)) in TEST_F() 37 .Input(FakeInput(DT_FLOAT)) in TEST_F() 38 .Input(FakeInput(DT_FLOAT)) in TEST_F() 68 .Input(FakeInput(DT_FLOAT)) in TEST_F() 69 .Input(FakeInput(DT_FLOAT)) in TEST_F() 70 .Input(FakeInput(DT_FLOAT)) in TEST_F() 71 .Input(FakeInput(DT_FLOAT)) in TEST_F() 72 .Input(FakeInput(DT_FLOAT)) in TEST_F() [all …]
|
D | mkl_quantized_concat_op_test.cc | 59 .Input(FakeInput(dtype)) // Input in ConvertMKL2TF() 60 .Input(FakeInput(DT_UINT8)) // MKL second tensor in ConvertMKL2TF() 92 .Input(FakeInput(2, DT_QUINT8)) in TestSmall8Bit() 93 .Input(FakeInput(DT_INT32)) in TestSmall8Bit() 94 .Input(FakeInput(2, DT_FLOAT)) in TestSmall8Bit() 95 .Input(FakeInput(2, DT_FLOAT)) in TestSmall8Bit() 96 .Input(FakeInput(2, DT_UINT8)) // MKL second tensor in TestSmall8Bit() 97 .Input(FakeInput(DT_UINT8)) // MKL second tensor in TestSmall8Bit() 98 .Input(FakeInput(2, DT_UINT8)) // MKL second tensor in TestSmall8Bit() 99 .Input(FakeInput(2, DT_UINT8)) // MKL second tensor in TestSmall8Bit() [all …]
|
D | mkl_quantized_conv_ops_test.cc | 55 .Input(FakeInput(dtype)) // Input in ConvertMklToTF() 56 .Input(FakeInput(DT_UINT8)) // MKL metadata tensor in ConvertMklToTF() 75 .Input(FakeInput(DT_QUINT8)) // Input in ConfigureQuantizedConv2D() 76 .Input(FakeInput(DT_QINT8)) // Filter in ConfigureQuantizedConv2D() 77 .Input(FakeInput(DT_FLOAT)) // Min input in ConfigureQuantizedConv2D() 78 .Input(FakeInput(DT_FLOAT)) // Max input in ConfigureQuantizedConv2D() 79 .Input(FakeInput(DT_FLOAT)) // Min filter in ConfigureQuantizedConv2D() 80 .Input(FakeInput(DT_FLOAT)) // Max filter in ConfigureQuantizedConv2D() 82 .Input(FakeInput(DT_UINT8)) in ConfigureQuantizedConv2D() 83 .Input(FakeInput(DT_UINT8)) in ConfigureQuantizedConv2D() [all …]
|
D | batch_norm_op_test.cc | 37 .Input(FakeInput(DT_FLOAT)) in TEST_F() 38 .Input(FakeInput(DT_FLOAT)) in TEST_F() 39 .Input(FakeInput(DT_FLOAT)) in TEST_F() 40 .Input(FakeInput(DT_FLOAT)) in TEST_F() 41 .Input(FakeInput(DT_FLOAT)) in TEST_F() 64 .Input(FakeInput(DT_HALF)) in TEST_F() 65 .Input(FakeInput(DT_HALF)) in TEST_F() 66 .Input(FakeInput(DT_HALF)) in TEST_F() 67 .Input(FakeInput(DT_HALF)) in TEST_F() 68 .Input(FakeInput(DT_HALF)) in TEST_F()
|
D | mkl_quantized_pooling_ops_test.cc | 47 .Input(FakeInput(dtype)) // Input in ConvertMKL2TF() 48 .Input(FakeInput(DT_UINT8)) // Mkl second tensor in ConvertMKL2TF() 68 .Input(FakeInput(DT_QUINT8)) in TEST_F() 69 .Input(FakeInput(DT_FLOAT)) in TEST_F() 70 .Input(FakeInput(DT_FLOAT)) in TEST_F() 71 .Input(FakeInput(DT_UINT8)) // MKl second tensor in TEST_F() 72 .Input(FakeInput(DT_UINT8)) // MKl second tensor in TEST_F() 73 .Input(FakeInput(DT_UINT8)) // MKl second tensor in TEST_F() 137 .Input(FakeInput(DT_QUINT8)) in TEST_F() 138 .Input(FakeInput(DT_FLOAT)) in TEST_F() [all …]
|
D | quantized_bias_add_op_test.cc | 42 .Input(FakeInput(DT_QUINT8)) in TEST_F() 43 .Input(FakeInput(DT_QUINT8)) in TEST_F() 44 .Input(FakeInput(DT_FLOAT)) in TEST_F() 45 .Input(FakeInput(DT_FLOAT)) in TEST_F() 46 .Input(FakeInput(DT_FLOAT)) in TEST_F() 47 .Input(FakeInput(DT_FLOAT)) in TEST_F() 92 .Input(FakeInput(DT_QUINT8)) in TEST_F() 93 .Input(FakeInput(DT_QUINT8)) in TEST_F() 94 .Input(FakeInput(DT_FLOAT)) in TEST_F() 95 .Input(FakeInput(DT_FLOAT)) in TEST_F() [all …]
|
D | mkl_requantize_ops_test.cc | 134 .Input(FakeInput(DT_QINT32)) in TEST_F() 135 .Input(FakeInput(DT_FLOAT)) in TEST_F() 136 .Input(FakeInput(DT_FLOAT)) in TEST_F() 195 .Input(FakeInput(DT_QINT32)) in TEST_F() 196 .Input(FakeInput(DT_FLOAT)) in TEST_F() 197 .Input(FakeInput(DT_FLOAT)) in TEST_F() 255 .Input(FakeInput(DT_QINT32)) in TEST_F() 256 .Input(FakeInput(DT_FLOAT)) in TEST_F() 257 .Input(FakeInput(DT_FLOAT)) in TEST_F() 258 .Input(FakeInput(DT_FLOAT)) in TEST_F() [all …]
|
D | sparse_reduce_sum_op_test.cc | 36 .Input(FakeInput(DT_INT64)) in MakeOp() 37 .Input(FakeInput(value_type)) in MakeOp() 38 .Input(FakeInput(DT_INT64)) in MakeOp() 39 .Input(FakeInput(DT_INT32)) in MakeOp() 78 .Input(FakeInput(DT_INT64)) in MakeOp() 79 .Input(FakeInput(value_type)) in MakeOp() 80 .Input(FakeInput(DT_INT64)) in MakeOp() 81 .Input(FakeInput(DT_INT32)) in MakeOp()
|
D | mkl_fused_ops_test.cc | 52 .Input(FakeInput(dtype)) // Input in PerformConversion() 53 .Input(FakeInput(DT_UINT8)) // Mkl second tensor in PerformConversion() 179 .Input(FakeInput(dtype)) in RunMklFusedConv2DOp() 180 .Input(FakeInput(dtype)) in RunMklFusedConv2DOp() 181 .Input(FakeInput(num_args, dtype)) in RunMklFusedConv2DOp() 182 .Input(FakeInput(DT_UINT8)) in RunMklFusedConv2DOp() 183 .Input(FakeInput(DT_UINT8)) in RunMklFusedConv2DOp() 184 .Input(FakeInput(num_args, DT_UINT8)) in RunMklFusedConv2DOp() 318 .Input(FakeInput(dtype)) // Input in Run() 319 .Input(FakeInput(dtype)) // Filter in Run() [all …]
|
D | adjust_contrast_op_test.cc | 36 .Input(FakeInput(DT_FLOAT)) in TEST_F() 37 .Input(FakeInput(DT_FLOAT)) in TEST_F() 51 .Input(FakeInput(DT_FLOAT)) in TEST_F() 52 .Input(FakeInput(DT_FLOAT)) in TEST_F() 68 .Input(FakeInput(DT_FLOAT)) in TEST_F() 69 .Input(FakeInput(DT_FLOAT)) in TEST_F()
|
D | quantized_activation_ops_test.cc | 40 .Input(FakeInput(DT_QUINT8)) in TEST_F() 41 .Input(FakeInput(DT_FLOAT)) in TEST_F() 42 .Input(FakeInput(DT_FLOAT)) in TEST_F() 71 .Input(FakeInput(DT_QUINT8)) in TEST_F() 72 .Input(FakeInput(DT_FLOAT)) in TEST_F() 73 .Input(FakeInput(DT_FLOAT)) in TEST_F()
|
D | quantized_concat_op_test.cc | 68 .Input(FakeInput(DT_INT32)) in TestSmall8Bit() 69 .Input(FakeInput(2, DT_QUINT8)) in TestSmall8Bit() 70 .Input(FakeInput(2, DT_FLOAT)) in TestSmall8Bit() 71 .Input(FakeInput(2, DT_FLOAT)) in TestSmall8Bit() 133 .Input(FakeInput(DT_INT32)) in TestSmall32Bit() 134 .Input(FakeInput(2, DT_QINT32)) in TestSmall32Bit() 135 .Input(FakeInput(2, DT_FLOAT)) in TestSmall32Bit() 136 .Input(FakeInput(2, DT_FLOAT)) in TestSmall32Bit() 196 .Input(FakeInput(DT_INT32)) in TestSecondDim8Bit() 197 .Input(FakeInput(2, DT_QUINT8)) in TestSecondDim8Bit() [all …]
|
D | requantize_op_test.cc | 35 .Input(FakeInput(DT_QINT32)) in ConfigureRequantize() 36 .Input(FakeInput(DT_FLOAT)) in ConfigureRequantize() 37 .Input(FakeInput(DT_FLOAT)) in ConfigureRequantize() 38 .Input(FakeInput(DT_FLOAT)) in ConfigureRequantize() 39 .Input(FakeInput(DT_FLOAT)) in ConfigureRequantize()
|
D | quantized_pooling_ops_test.cc | 42 .Input(FakeInput(DT_QUINT8)) in TEST_F() 43 .Input(FakeInput(DT_FLOAT)) in TEST_F() 44 .Input(FakeInput(DT_FLOAT)) in TEST_F() 87 .Input(FakeInput(DT_QUINT8)) in TEST_F() 88 .Input(FakeInput(DT_FLOAT)) in TEST_F() 89 .Input(FakeInput(DT_FLOAT)) in TEST_F()
|
/external/tensorflow/tensorflow/core/framework/ |
D | node_def_builder_test.cc | 127 ExpectSuccess(Builder().Input(FakeInput()), {DT_INT32}, {DT_FLOAT}, R"proto( in TEST_F() 130 ExpectSuccess(Builder().Input(FakeInput(DT_INT32)), {DT_INT32}, {DT_FLOAT}, in TEST_F() 134 ExpectSuccess(Builder().Input(FakeInput(DT_INT32_REF)), {DT_INT32}, in TEST_F() 139 Builder().ControlInput("x").Input(FakeInput()).ControlInput("y"), in TEST_F() 144 ExpectSuccess(Builder().Input(FakeInput()).Device("ddd"), {DT_INT32}, in TEST_F() 160 TF_EXPECT_OK(builder.Input(FakeInput()).Finalize(nullptr)); in TEST_F() 170 builder.Input(FakeInput()); in TEST_F() 175 builder.Input(FakeInput(DT_INT32)); in TEST_F() 187 ExpectFailure(Builder().Input(FakeInput(3, DT_FLOAT)), in TEST_F() 190 ExpectFailure(Builder().Input(FakeInput(3)), in TEST_F() [all …]
|
D | fake_input.h | 29 FakeInputFunctor FakeInput(); // Infer everything 30 FakeInputFunctor FakeInput(DataType dt); 31 FakeInputFunctor FakeInput(int n); // List of length n 32 FakeInputFunctor FakeInput(int n, DataType dt); 33 FakeInputFunctor FakeInput(DataTypeSlice dts); 34 inline FakeInputFunctor FakeInput(std::initializer_list<DataType> dts) { in FakeInput() function 35 return FakeInput(DataTypeSlice(dts)); in FakeInput()
|
D | node_def_util_test.cc | 380 NodeDefBuilder("simple", &op_def).Input(FakeInput()).Input(FakeInput())); in TEST() 401 NodeDefBuilder("simple", &op_def).Input(FakeInput()).Input(FakeInput())); in TEST() 423 NodeDefBuilder("simple", &op_def).Input(FakeInput()).Input(FakeInput())); in TEST() 443 .Input(FakeInput(DT_INT32)) in TEST() 444 .Input(FakeInput(DT_INT32))); in TEST() 452 .Input(FakeInput(DT_BOOL)) in TEST() 453 .Input(FakeInput(DT_BOOL))); in TEST() 473 .Input(FakeInput(4, DT_INT32)) in TEST() 474 .Input(FakeInput(4, DT_FLOAT)) in TEST() 486 .Input(FakeInput(2, DT_INT32)) in TEST() [all …]
|
D | op_compatibility_test.cc | 204 .Input(FakeInput()) in TEST_F() 205 .Input(FakeInput(DT_FLOAT)) in TEST_F() 206 .Input(FakeInput(3)) in TEST_F() 207 .Input(FakeInput(3, DT_FLOAT)) in TEST_F() 208 .Input(FakeInput(2, DT_BOOL)) in TEST_F() 300 .Input(FakeInput()) in TEST_F() 322 .Input(FakeInput()) in TEST_F() 345 .Input(FakeInput()) in TEST_F() 367 .Input(FakeInput()) in TEST_F() 390 .Input(FakeInput(DT_INT32)) in TEST_F() [all …]
|
D | memory_types_test.cc | 58 .Input(FakeInput()) in TEST() 59 .Input(FakeInput(DT_BOOL)) in TEST() 60 .Input(FakeInput(3)) in TEST() 61 .Input(FakeInput({DT_INT32, DT_FLOAT, DT_INT32})) in TEST() 62 .Input(FakeInput({DT_RESOURCE, DT_STRING, DT_RESOURCE})) in TEST()
|