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

  /external/tensorflow/tensorflow/core/kernels/
gather_op.cc 55 const Tensor& axis_tensor = c->input(2); variable
56 OP_REQUIRES(c, TensorShapeUtils::IsScalar(axis_tensor.shape()),
59 if (axis_tensor.dtype() == DT_INT32) {
60 axis = axis_tensor.scalar<int32>()();
61 } else if (axis_tensor.dtype() == DT_INT64) {
62 axis = axis_tensor.scalar<int64>()();
unique_op.cc 56 const Tensor& axis_tensor = context->input(1); variable
57 OP_REQUIRES(context, TensorShapeUtils::IsVector(axis_tensor.shape()),
60 context, axis_tensor.NumElements() <= 1,
63 if (axis_tensor.NumElements() == 0) {
68 (axis_tensor.dtype() == DT_INT32 ||
69 axis_tensor.dtype() == DT_INT64),
72 axis_tensor.dtype()));
73 if (axis_tensor.dtype() == DT_INT32) {
74 axis = internal::SubtleMustCopy(axis_tensor.scalar<int32>()());
76 axis = internal::SubtleMustCopy(axis_tensor.scalar<int64>()())
    [all...]

Completed in 177 milliseconds