Searched defs:output_start_indices (Results 1 - 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
H A Dsparse_cross_op.cc142 OutputUpdater(const std::vector<int64>& output_start_indices, argument
144 : output_start_indices_(output_start_indices),
324 std::vector<int64> output_start_indices(batch_size);
326 &shape_out, &output_start_indices);
329 output_start_indices, indices_out, values_out);
501 // It also output_start_indices which contains the start indices for each
508 std::vector<int64>* output_start_indices) {
514 (*output_start_indices)[b] = cross_count_total;
503 CreateOutputTensors( const std::vector<std::unique_ptr<ColumnInterface<InternalType>>>& columns, int64 batch_size, OpKernelContext* context, Tensor** indices_out, Tensor** values_out, Tensor** shape_out, std::vector<int64>* output_start_indices) argument
/external/tensorflow/tensorflow/contrib/layers/kernels/
H A Dsparse_feature_cross_kernel.cc142 OutputUpdater(const std::vector<int64>& output_start_indices, argument
144 : output_start_indices_(output_start_indices),
367 std::vector<int64> output_start_indices(batch_size);
369 &shape_out, &output_start_indices);
372 updater(output_start_indices, indices_out, values_out);
544 // It also output_start_indices which contains the start indices for each
551 std::vector<int64>* output_start_indices) {
557 (*output_start_indices)[b] = cross_count_total;
546 CreateOutputTensors( const std::vector<std::unique_ptr<ColumnInterface<InternalType>>>& columns, int64 batch_size, OpKernelContext* context, Tensor** indices_out, Tensor** values_out, Tensor** shape_out, std::vector<int64>* output_start_indices) argument

Completed in 66 milliseconds