Searched refs:start_batch (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | topk_op.cc | 139 auto SortIndices = [&](int64 start_batch, int64 limit_batch) { in Compute() argument
140 for (int32 b = start_batch; b < limit_batch; ++b) { in Compute()
|
| D | parameterized_truncated_normal_op.cc | 73 kStdDevsInsideBoundsToUseRandnSampler](int64 start_batch, in operator ()()
82 gen_copy.Skip(start_batch * 2 * kMaxIterations * (samples_per_batch + 3) / in operator ()()
94 for (int64 b = start_batch; b < limit_batch; ++b) { in operator ()()
|
| /external/grpc-grpc/src/objective-c/GRPCClient/ |
| D | GRPCCall.m | 59 // - start_batch
62 // start_batch with a SEND_MESSAGE argument can only be called after the
67 // Similarly, start_batch with a RECV_MESSAGE argument can only be called after
70 // start_batch is enqueued once upon receiving the OP_COMPLETE event for the
|
| /external/rust/crates/grpcio-sys/grpc/src/core/lib/surface/ |
| D | call.cc | 106 grpc_closure start_batch; member
689 grpc_closure start_batch; member
721 execute_batch(c, op, &state->start_batch); in cancel_with_error()
1936 execute_batch(call, stream_op, &bctl->start_batch); in call_start_batch()
|
| /external/grpc-grpc/src/core/lib/surface/ |
| D | call.cc | 96 grpc_closure start_batch; member
646 grpc_closure start_batch; member
679 execute_batch(c, op, &state->start_batch); in cancel_with_error()
1844 execute_batch(call, stream_op, &bctl->start_batch); in call_start_batch()
|
| /external/grpc-grpc/doc/ |
| D | combiner-explainer.md | 94 `start_batch(..., tag)` and then work only gets activated by somebody
|
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| D | optimized_ops.h | 3651 int start_batch, int end_batch) { in SoftmaxImpl() argument
3656 const MatrixMap<const float> in_mat(input_data + logit_size * start_batch, in SoftmaxImpl()
3657 logit_size, end_batch - start_batch); in SoftmaxImpl()
3658 MatrixMap<float> out_mat(output_data + logit_size * start_batch, logit_size, in SoftmaxImpl()
3659 end_batch - start_batch); in SoftmaxImpl()
3676 int start_batch, int end_batch) in SoftmaxWorkerTask()
3682 start_batch(start_batch), in SoftmaxWorkerTask()
3686 start_batch, end_batch); in Run()
3695 int start_batch; member
|