Searched refs:batch_index_t (Results 1 – 1 of 1) sorted by relevance
883 const Tensor& batch_index_t = context->input(1); in Compute() local885 if (batch_index_t.shape().dim_size(0) > data_t.shape().dim_size(0)) { in Compute()890 "; Got: ", batch_index_t.shape().dim_size(0), "."); in Compute()892 if (batch_index_t.shape().dim_size(1) != 3) { in Compute()896 batch_index_t.shape().dim_size(1), "."); in Compute()900 const bool nonempty_input = batch_index_t.dim_size(0) > 0; in Compute()910 batch_index_t.shaped<int64, 2>({batch_index_t.dim_size(0), 3}); in Compute()911 for (int i = 0; i < batch_index_t.dim_size(0); ++i) { in Compute()1102 const Tensor& batch_index_t = context->input(1); in OutputBatch() local1104 batch_index_t.shaped<int64, 2>({batch_index_t.dim_size(0), 3}); in OutputBatch()[all …]