Searched refs:GetEngine (Results 1 – 18 of 18) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_avgpooling_op.cc | 132 fwd_cpu_stream.reset(CreateStream(context, pooling_fwd->GetEngine())); in Compute()
253 bwd_cpu_stream.reset(CreateStream(context, pooling_bwd->GetEngine())); in Compute()
|
| D | mkl_maxpooling_op.cc | 156 fwd_cpu_stream.reset(CreateStream(context, pooling_fwd->GetEngine())); in Compute()
307 bwd_cpu_stream.reset(CreateStream(context, pooling_bwd->GetEngine())); in Compute()
|
| D | mkl_batch_matmul_op.cc | 147 cpu_stream.reset(CreateStream(ctx, matmul_prim->GetEngine())); in Compute()
|
| D | mkl_transpose_op.cc | 147 transpose_stream.reset(CreateStream(context, prim->GetEngine())); in MKLTransposeND()
|
| D | mkl_softmax_op.cc | 301 fwd_cpu_stream.reset(CreateStream(context, softmax_fwd->GetEngine())); in Compute()
|
| D | mkl_matmul_op_fused.cc | 252 cpu_stream.reset(CreateStream(ctx, matmul_prim->GetEngine())); in Compute()
|
| D | mkl_quantize_op.cc | 477 cpu_stream.reset(CreateStream(ctx, reorder_prim->GetEngine())); in Compute()
|
| D | mkl_slice_op.cc | 456 slice_stream.reset(CreateStream(context, reorder_prim->GetEngine())); in ComputeMklSlice()
|
| D | mkl_conv_grad_input_ops.cc | 485 bwd_cpu_stream.reset(CreateStream(context, conv_bwd_input->GetEngine())); in Compute()
|
| D | mkl_qmatmul_op.cc | 292 cpu_stream.reset(CreateStream(context, matmul_fwd->GetEngine())); in Compute()
|
| D | mkl_relu_op.cc | 482 fwd_cpu_stream.reset(CreateStream(context, eltwise_fwd->GetEngine())); in Compute()
688 bwd_cpu_stream.reset(CreateStream(context, eltwise_bwd->GetEngine())); in Compute()
|
| D | mkl_conv_grad_filter_ops.cc | 588 bwd_cpu_stream.reset(CreateStream(context, conv_bwd_filter->GetEngine())); in Compute()
|
| D | mkl_fused_batch_norm_op.cc | 869 fwd_cpu_stream.reset(CreateStream(context, bn_fwd->GetEngine())); in Compute()
1275 bwd_cpu_stream.reset(CreateStream(context, bn_bwd->GetEngine())); in Compute()
|
| D | mkl_matmul_ops_common.h | 721 cpu_stream.reset(CreateStream(ctx, matmul_prim->GetEngine()));
|
| D | mkl_concat_op.cc | 772 fwd_cpu_stream.reset(CreateStream(context, concat_fwd->GetEngine())); in Compute()
|
| D | mkl_conv_ops.cc | 727 fwd_cpu_stream.reset(CreateStream(context, conv_fwd->GetEngine())); in Compute()
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/kernels/ |
| D | trt_engine_op.cc | 137 StatusOr<std::pair<EngineContext*, int>> GetEngine(
698 GetEngine(input_concrete_shapes, ctx, cache_res); in ComputeAsync()
862 StatusOr<std::pair<EngineContext*, int>> TRTEngineOp::GetEngine( in GetEngine() function in tensorflow::tensorrt::TRTEngineOp
|
| /external/tensorflow/tensorflow/core/util/ |
| D | mkl_util.h | 1498 cpu_stream.reset(CreateStream(context, prim->GetEngine()));
1559 cpu_stream.reset(CreateStream(context, prim->GetEngine()));
1665 cpu_stream.reset(CreateStream(ctx, prim->GetEngine()));
1679 const engine& GetEngine() { return cpu_engine_; } in GetEngine() function
|