Searched refs:input_generator (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/lite/testing/kernel_test/ |
| D | input_generator_test.cc | 30 InputGenerator input_generator; in TEST() local
31 ASSERT_EQ(input_generator.LoadModel( in TEST()
37 InputGenerator input_generator; in TEST() local
38 ASSERT_EQ(input_generator.ReadInputsFromFile( in TEST()
48 ASSERT_EQ(input_generator.GetInputs(), inputs); in TEST()
51 ASSERT_EQ(input_generator.WriteInputsToFile(output_filename), kTfLiteOk); in TEST()
60 InputGenerator input_generator; in TEST() local
61 ASSERT_EQ(input_generator.LoadModel( in TEST()
64 input_generator.GenerateInput("UNIFORM"); in TEST()
65 auto inputs = input_generator.GetInputs(); in TEST()
[all …]
|
| D | util.h | 69 InputGenerator input_generator; in RunKernelTest() local
72 TF_LITE_ENSURE_STATUS(input_generator.LoadModel(options.tflite_model)); in RunKernelTest()
74 input_generator.GenerateInput(options.input_distribution)); in RunKernelTest()
77 input_generator.ReadInputsFromFile(options.read_input_from_file)); in RunKernelTest()
84 auto inputs = input_generator.GetInputs(); in RunKernelTest()
98 input_generator.WriteInputsToFile(options.dump_input_to_file)); in RunKernelTest()
|
| D | BUILD | 17 ":input_generator",
68 name = "input_generator",
69 srcs = ["input_generator.cc"],
70 hdrs = ["input_generator.h"],
93 ":input_generator",
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | ir_emitter.h | 124 const llvm_ir::ElementGenerator& input_generator,
129 const llvm_ir::ElementGenerator& input_generator,
135 const llvm_ir::ElementGenerator& input_generator,
|
| D | ir_emitter.cc | 642 const llvm_ir::ElementGenerator& input_generator, in EmitElementalReduceWindow() argument
718 input_generator(input_index)); in EmitElementalReduceWindow()
952 const llvm_ir::ElementGenerator& input_generator, in EmitElementalConvolution() argument
1080 input_generator(input_index)); in EmitElementalConvolution()
1816 const llvm_ir::ElementGenerator& input_generator, in EmitElementalReduce() argument
1862 input_generator(input_index)); in EmitElementalReduce()
|