HomeSort by relevance Sort by last modified time
    Searched refs:input_data_type (Results 1 - 5 of 5) sorted by null

  /external/tensorflow/tensorflow/core/kernels/
mkl_tfconv_op.h 84 DataType input_data_type = op_kernel->input_type(input_number); local
86 CHECK_EQ(op_data_type, input_data_type);
147 DataType input_data_type = op_kernel->input_type(input_number); local
149 CHECK_EQ(op_data_type, input_data_type);
  /external/tensorflow/tensorflow/contrib/lite/toco/
model_cmdline_flags.cc 75 Flag("input_data_type", parsed_flags.input_data_type.bind(),
76 parsed_flags.input_data_type.default_value(),
265 if (parsed_model_flags.input_data_type.specified()) {
268 QCHECK(IODataType_Parse(parsed_model_flags.input_data_type.value(), &type));
args.h 197 Arg<string> input_data_type; member in struct:toco::ParsedModelFlags
  /external/tensorflow/tensorflow/stream_executor/cuda/
cuda_dnn.h 621 Stream* stream, dnn::DataType input_data_type,
    [all...]
cuda_dnn.cc     [all...]

Completed in 113 milliseconds