Home
last modified time | relevance | path

Searched refs:GetEngine (Results 1 – 25 of 26) sorted by relevance

12

/external/googleapis/google/cloud/discoveryengine/v1/
Dengine.proto63 // [EngineService.GetEngine][google.cloud.discoveryengine.v1.EngineService.GetEngine]
95 // [EngineService.GetEngine][google.cloud.discoveryengine.v1.EngineService.GetEngine]
110 // [EngineService.GetEngine][google.cloud.discoveryengine.v1.EngineService.GetEngine]
Dengine_service.proto80 rpc GetEngine(GetEngineRequest) returns (Engine) { rpc
173 // [EngineService.GetEngine][google.cloud.discoveryengine.v1.EngineService.GetEngine]
/external/googleapis/google/cloud/discoveryengine/v1beta/
Dengine.proto63 // [EngineService.GetEngine][google.cloud.discoveryengine.v1beta.EngineService.GetEngine]
95 // [EngineService.GetEngine][google.cloud.discoveryengine.v1beta.EngineService.GetEngine]
110 // [EngineService.GetEngine][google.cloud.discoveryengine.v1beta.EngineService.GetEngine]
Dengine_service.proto80 rpc GetEngine(GetEngineRequest) returns (Engine) { rpc
211 // [EngineService.GetEngine][google.cloud.discoveryengine.v1beta.EngineService.GetEngine]
/external/googleapis/google/cloud/discoveryengine/v1alpha/
Dengine.proto138 // [EngineService.GetEngine][google.cloud.discoveryengine.v1alpha.EngineService.GetEngine]
170 // [EngineService.GetEngine][google.cloud.discoveryengine.v1alpha.EngineService.GetEngine]
185 // [EngineService.GetEngine][google.cloud.discoveryengine.v1alpha.EngineService.GetEngine]
Dengine_service.proto80 rpc GetEngine(GetEngineRequest) returns (Engine) { rpc
211 // [EngineService.GetEngine][google.cloud.discoveryengine.v1alpha.EngineService.GetEngine]
/external/tensorflow/tensorflow/core/kernels/mkl/
Dmkl_avgpooling_op.cc130 fwd_cpu_stream.reset(CreateStream(&eigen_tp, pooling_fwd->GetEngine())); in Compute()
264 bwd_cpu_stream.reset(CreateStream(&eigen_tp, pooling_bwd->GetEngine())); in Compute()
Dmkl_transpose_op.cc88 transpose_stream.reset(CreateStream(&eigen_tp, prim->GetEngine())); in MKLTransposeND()
Dmkl_maxpooling_op.cc152 fwd_cpu_stream.reset(CreateStream(&eigen_tp, pooling_fwd->GetEngine())); in Compute()
314 bwd_cpu_stream.reset(CreateStream(&eigen_tp, pooling_bwd->GetEngine())); in Compute()
Dmkl_softmax_op.cc325 fwd_cpu_stream.reset(CreateStream(&eigen_tp, softmax_fwd->GetEngine())); in Compute()
Dmkl_einsum_op.cc123 cpu_stream.reset(CreateStream(&eigen_tp, matmul_prim->GetEngine())); in MKLContractOperands()
Dmkl_batch_matmul_op.cc163 cpu_stream.reset(CreateStream(&eigen_tp, matmul_prim->GetEngine())); in Compute()
Dmkl_eltwise_activation_base_op.h103 ExecutePrimitive(net, &net_args, GetEngine(), op_context); in Execute()
Dmkl_matmul_op_fused.cc268 cpu_stream.reset(CreateStream(&eigen_tp, matmul_prim->GetEngine())); in Compute()
Dmkl_quantize_op.cc487 cpu_stream.reset(CreateStream(&eigen_tp, reorder_prim->GetEngine())); in Compute()
Dmkl_slice_op.cc464 slice_stream.reset(CreateStream(&eigen_tp, reorder_prim->GetEngine())); in ComputeMklSlice()
Dmkl_qmatmul_op.cc295 cpu_stream.reset(CreateStream(&eigen_tp, matmul_fwd->GetEngine())); in Compute()
Dmkl_relu_op.cc484 fwd_cpu_stream.reset(CreateStream(&eigen_tp, eltwise_fwd->GetEngine())); in Compute()
691 bwd_cpu_stream.reset(CreateStream(&eigen_tp, eltwise_bwd->GetEngine())); in Compute()
Dmkl_conv_grad_input_ops.cc488 CreateStream(&eigen_tp, conv_bwd_input->GetEngine())); in Compute()
Dmkl_conv_grad_filter_ops.cc617 CreateStream(&eigen_tp, conv_bwd_filter->GetEngine())); in Compute()
Dmkl_fused_batch_norm_op.cc931 fwd_cpu_stream.reset(CreateStream(&eigen_tp, bn_fwd->GetEngine())); in Compute()
1368 bwd_cpu_stream.reset(CreateStream(&eigen_tp, bn_bwd->GetEngine())); in Compute()
Dmkl_concat_op.cc787 CreateStream(&eigen_tp, concat_fwd->GetEngine())); in Compute()
Dmkl_matmul_ops_common.h917 cpu_stream.reset(CreateStream(&eigen_tp, matmul_prim->GetEngine()));
/external/tensorflow/tensorflow/core/util/
Dmkl_util.h1546 cpu_stream.reset(CreateStream(&eigen_tp, prim->GetEngine()));
1548 cpu_stream.reset(CreateStream(nullptr, prim->GetEngine()));
1613 cpu_stream.reset(CreateStream(&eigen_tp, prim->GetEngine()));
1615 cpu_stream.reset(CreateStream(nullptr, prim->GetEngine()));
1724 cpu_stream.reset(CreateStream(&eigen_tp, prim->GetEngine()));
1726 cpu_stream.reset(CreateStream(nullptr, prim->GetEngine()));
1741 const engine& GetEngine() { return cpu_engine_; } in GetEngine() function
/external/tensorflow/tensorflow/compiler/tf2tensorrt/kernels/
Dtrt_engine_op.cc187 StatusOr<std::pair<EngineContext*, int>> GetEngine(
897 GetEngine(input_concrete_shapes, ctx, cache_res); in ComputeAsync()
1089 StatusOr<std::pair<EngineContext*, int>> TRTEngineOp::GetEngine( in GetEngine() function in tensorflow::tensorrt::TRTEngineOp

12