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

  /external/tensorflow/tensorflow/core/kernels/
serialize_sparse_op.cc 76 auto serialized_sparse_t = serialized_sparse.vec<T>(); variable
77 OP_REQUIRES_OK(context, Serialize(*input_indices, &serialized_sparse_t(0)));
78 OP_REQUIRES_OK(context, Serialize(*input_values, &serialized_sparse_t(1)));
79 OP_REQUIRES_OK(context, Serialize(*input_shape, &serialized_sparse_t(2)));
181 auto serialized_sparse_t = serialized_sparse.matrix<U>(); variable
190 serialized_sparse_t.template chip<1>(0).setConstant(serialized_indices);
196 serialized_sparse_t.template chip<1>(1).setConstant(serialized_values);
203 serialized_sparse_t.template chip<1>(2).setConstant(serialized_shape);
233 context, this->Serialize(output_indices, &serialized_sparse_t(b, 0)));
235 context, this->Serialize(output_values, &serialized_sparse_t(b, 1)))
329 const auto& serialized_sparse_t = serialized_sparse.vec<T>(); variable
351 const auto& serialized_sparse_t = serialized_sparse.flat_inner_dims<T, 2>(); variable
    [all...]
  /external/tensorflow/tensorflow/core/kernels/data/
dense_to_sparse_batch_dataset_op.cc 259 auto serialized_sparse_t = serialized_sparse.vec<Variant>(); variable
260 serialized_sparse_t(0) = std::move(indices);
261 serialized_sparse_t(1) = std::move(values);
262 serialized_sparse_t(2) = std::move(dense_shape);

Completed in 751 milliseconds