Searched refs:gather_request (Results 1 - 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
H A Duser_computation.cc319 const GatherRequest& gather_request) {
323 LookUpRequest(gather_request.input()));
325 LookUpRequest(gather_request.gather_indices()));
331 gather_request.dimension_numbers(),
332 AsInt64Slice(gather_request.window_bounds())));
340 *request.mutable_request()->mutable_gather_request() = gather_request;
344 << gather_request.ShortDebugString();
2053 request.request().gather_request().input(),
2056 request.request().gather_request().gather_indices(),
2764 const GatherRequest& gather_request local
318 AddGatherInstruction( const GatherRequest& gather_request) argument
3504 const GatherRequest& gather_request = request.request().gather_request(); local
[all...]
H A Duser_computation.h247 const GatherRequest& gather_request);
H A Dservice.cc1450 handle_status = computation->AddGatherInstruction(arg->gather_request());
/external/tensorflow/tensorflow/compiler/xla/client/
H A Dcomputation_builder.cc1243 GatherRequest* gather_request = op_request.mutable_gather_request(); local
1244 *gather_request->mutable_input() = input;
1245 *gather_request->mutable_gather_indices() = gather_indices;
1246 *gather_request->mutable_dimension_numbers() = dimension_numbers;
1248 gather_request->add_window_bounds(window_bound);
/external/tensorflow/tensorflow/compiler/xla/
H A Dxla_data.proto1009 GatherRequest gather_request = 46;

Completed in 414 milliseconds