Searched defs:output_start_indices (Results 1 - 2 of 2) sorted by relevance
/external/tensorflow/tensorflow/core/kernels/ |
H A D | sparse_cross_op.cc | 142 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 D | sparse_feature_cross_kernel.cc | 142 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