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

  /external/tensorflow/tensorflow/core/kernels/
sparse_cross_op.cc 301 OpInputList indices_list_in; variable
302 OP_REQUIRES_OK(context, context->input_list("indices", &indices_list_in));
311 ValidateInput(context, indices_list_in, values_list_in, shapes_list_in,
315 GenerateColumnsFromInput(indices_list_in, values_list_in,
344 const int kCostPerUnit = 5000 * indices_list_in.size();
352 const OpInputList& indices_list_in,
356 const auto size = indices_list_in.size();
357 // Validates indices_list_in OpInputList.
360 context, TensorShapeUtils::IsMatrix(indices_list_in[i].shape()),
363 indices_list_in[i].shape().DebugString(), " at position ", i))
    [all...]
  /external/tensorflow/tensorflow/contrib/layers/kernels/
sparse_feature_cross_kernel.cc 345 OpInputList indices_list_in; variable
346 OP_REQUIRES_OK(context, context->input_list("indices", &indices_list_in));
354 ValidateInput(context, indices_list_in, values_list_in, shapes_list_in,
358 GenerateColumnsFromInput(indices_list_in, values_list_in,
387 const int kCostPerUnit = 5000 * indices_list_in.size();
395 const OpInputList& indices_list_in,
399 const auto size = indices_list_in.size();
400 // Validates indices_list_in OpInputList.
403 context, TensorShapeUtils::IsMatrix(indices_list_in[i].shape()),
406 indices_list_in[i].shape().DebugString(), " at position ", i))
    [all...]

Completed in 398 milliseconds