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

  /external/tensorflow/tensorflow/core/kernels/
mkl_tfconv_op.h 85 DataType output_data_type = op_kernel->output_type(input_number); local
87 CHECK_EQ(op_data_type, output_data_type);
148 DataType output_data_type = op_kernel->output_type(input_number); local
150 CHECK_EQ(op_data_type, output_data_type);
  /external/tensorflow/tensorflow/contrib/lite/toco/graph_transformations/
propagate_array_data_types.cc 83 model->GetArray(op->outputs[0]).data_type = argmax_op->output_data_type;
resolve_constant_binary.cc 170 const auto output_data_type = local
174 output_data_type == OutputDataType) { \
  /external/tensorflow/tensorflow/contrib/lite/toco/
model.h 1326 ArrayDataType output_data_type = ArrayDataType::kInt64; member in struct:toco::ArgMaxOperator
    [all...]
import_tensorflow.cc     [all...]

Completed in 80 milliseconds