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

  /external/tensorflow/tensorflow/lite/kernels/
one_hot_test.cc 34 T off_value = 0, TensorType indices_type = TensorType_INT32) {
35 indices_ = AddInput(indices_type);
  /external/tensorflow/tensorflow/compiler/tf2xla/kernels/
softmax_op.cc 205 const DataType indices_type = input_type(1); variable
224 xla::And(xla::Le(XlaHelpers::Zero(builder, indices_type), indices),
226 builder, indices_type, depth))),
gather_op.cc 204 DataType indices_type = context->input_type(1); variable
228 indices_type, builder, &gather));
  /external/tensorflow/tensorflow/compiler/xla/service/
algebraic_simplifier.cc     [all...]
  /external/tensorflow/tensorflow/lite/toco/
export_tensorflow.cc 1791 const tensorflow::DataType indices_type = local
    [all...]

Completed in 361 milliseconds