| /external/tensorflow/tensorflow/python/autograph/tests/ |
| D | loop_distributed_test.py | 25 def no_vars_loop(strat, iterable): argument
27 tf.print(strat.reduce('SUM', pr, axis=0))
30 def single_var_loop(strat, iterable): argument
34 s = s * 10 + strat.reduce('SUM', pr, axis=0)
38 def loop_with_break(strat, iterable): argument
41 if strat.reduce('SUM', pr, axis=0) % 5 == 0:
43 s = s * 10 + strat.reduce('SUM', pr, axis=0)
47 def loop_with_continue(strat, iterable): argument
50 if strat.reduce('SUM', pr, axis=0) % 2 == 0:
52 s = s * 10 + strat.reduce('SUM', pr, axis=0)
[all …]
|
| /external/ComputeLibrary/src/core/NEON/kernels/arm_conv/depthwise/ |
| D | depthwise_fp32.cpp | 133 auto strat = new sme2_fp32bf16fp32_planar_3x3_s1_4rows_dot_za(args.cpu_info); in __anon7831a4000202() local
134 return new DepthwisePlanar<float>(strat, args); in __anon7831a4000202()
146 auto strat = new sme2_fp32bf16fp32_planar_3x3_s2_4rows_dot_za(args.cpu_info); in __anon7831a4000302() local
147 return new DepthwisePlanar<float>(strat, args); in __anon7831a4000302()
159 auto strat = new sme2_fp32bf16fp32_planar_5x5_s1_4rows_dot_za(args.cpu_info); in __anon7831a4000402() local
160 return new DepthwisePlanar<float>(strat, args); in __anon7831a4000402()
172 auto strat = new sme2_fp32bf16fp32_planar_5x5_s2_4rows_dot_za(args.cpu_info); in __anon7831a4000502() local
173 return new DepthwisePlanar<float>(strat, args); in __anon7831a4000502()
192 auto strat = new sme2_fp32_planar_3x3_s1_4rows_mla_za(args.cpu_info); in __anon7831a4000702() local
193 return new DepthwisePlanar<float>(strat, args); in __anon7831a4000702()
[all …]
|
| D | depthwise_u8q.cpp | 87 auto strat = new sme2_u8q_planar_3x3_s1_4rows_dot_za(args.cpu_info); in __anon842cbe830102() local
88 return new DepthwisePlanar<uint8_t>(strat, args, qp); in __anon842cbe830102()
100 auto strat = new sme2_u8q_planar_3x3_s2_4rows_dot_za(args.cpu_info); in __anon842cbe830202() local
101 return new DepthwisePlanar<uint8_t>(strat, args, qp); in __anon842cbe830202()
113 auto strat = new sme2_u8q_planar_5x5_s1_4rows_dot_za(args.cpu_info); in __anon842cbe830302() local
114 return new DepthwisePlanar<uint8_t>(strat, args, qp); in __anon842cbe830302()
126 auto strat = new sme2_u8q_planar_5x5_s2_4rows_dot_za(args.cpu_info); in __anon842cbe830402() local
127 return new DepthwisePlanar<uint8_t>(strat, args, qp); in __anon842cbe830402()
140 auto strat = new sve_u8q_nhwc_3x3_s1_output2x2_dot_depthfirst(args.cpu_info); in __anon842cbe830502() local
141 return new DepthwiseDepthfirst<uint8_t>(strat, args, qp); in __anon842cbe830502()
[all …]
|
| D | depthwise_s8q.cpp | 94 auto strat = new sme2_s8q_planar_3x3_s1_4rows_dot_za(args.cpu_info); in __anonea3845010202() local
95 return new DepthwisePlanar<int8_t>(strat, args, qp); in __anonea3845010202()
107 auto strat = new sme2_s8q_planar_3x3_s2_4rows_dot_za(args.cpu_info); in __anonea3845010302() local
108 return new DepthwisePlanar<int8_t>(strat, args, qp); in __anonea3845010302()
120 auto strat = new sme2_s8q_planar_5x5_s1_4rows_dot_za(args.cpu_info); in __anonea3845010402() local
121 return new DepthwisePlanar<int8_t>(strat, args, qp); in __anonea3845010402()
133 auto strat = new sme2_s8q_planar_5x5_s2_4rows_dot_za(args.cpu_info); in __anonea3845010502() local
134 return new DepthwisePlanar<int8_t>(strat, args, qp); in __anonea3845010502()
148 auto strat = new sve_s8qs_nhwc_3x3_s1_output2x2_dot_depthfirst(args.cpu_info); in __anonea3845010602() local
149 return new DepthwiseDepthfirst<int8_t>(strat, args, qp); in __anonea3845010602()
[all …]
|
| D | depthwise_fp16.cpp | 93 auto strat = new sve_fp16_nhwc_3x3_s1_output4x4_mla_depthfirst(args.cpu_info); in __anon73cfb4c20202() local
94 return new DepthwiseDepthfirst<__fp16>(strat, args); in __anon73cfb4c20202()
105 auto strat = new sve_fp16_nhwc_3x3_s1_output3x3_mla_depthfirst(args.cpu_info); in __anon73cfb4c20302() local
106 return new DepthwiseDepthfirst<__fp16>(strat, args); in __anon73cfb4c20302()
117 auto strat = new sve_fp16_nhwc_3x3_s1_output2x2_mla_depthfirst(args.cpu_info); in __anon73cfb4c20402() local
118 return new DepthwiseDepthfirst<__fp16>(strat, args); in __anon73cfb4c20402()
129 auto strat = new sve_fp16_nhwc_3x3_s2_output2x2_mla_depthfirst(args.cpu_info); in __anon73cfb4c20502() local
130 return new DepthwiseDepthfirst<__fp16>(strat, args); in __anon73cfb4c20502()
141 auto strat = new sve_fp16_nhwc_5x5_s1_output2x2_mla_depthfirst(args.cpu_info); in __anon73cfb4c20602() local
142 return new DepthwiseDepthfirst<__fp16>(strat, args); in __anon73cfb4c20602()
[all …]
|
| D | depthwise_u8s8u8q.cpp | 74 auto strat = new sme2_u8s8u8q_planar_3x3_s1_4rows_dot_za(args.cpu_info); in __anon3c4de31b0102() local
75 return new DepthwisePlanar<uint8_t, int8_t>(strat, args, qp); in __anon3c4de31b0102()
87 auto strat = new sme2_u8s8u8q_planar_3x3_s2_4rows_dot_za(args.cpu_info); in __anon3c4de31b0202() local
88 return new DepthwisePlanar<uint8_t, int8_t>(strat, args, qp); in __anon3c4de31b0202()
100 auto strat = new sme2_u8s8u8q_planar_5x5_s1_4rows_dot_za(args.cpu_info); in __anon3c4de31b0302() local
101 return new DepthwisePlanar<uint8_t, int8_t>(strat, args, qp); in __anon3c4de31b0302()
113 auto strat = new sme2_u8s8u8q_planar_5x5_s2_4rows_dot_za(args.cpu_info); in __anon3c4de31b0402() local
114 return new DepthwisePlanar<uint8_t, int8_t>(strat, args, qp); in __anon3c4de31b0402()
127 auto strat = new sve_u8s8u8q_nhwc_3x3_s1_output2x2_mla_depthfirst(args.cpu_info); in __anon3c4de31b0502() local
128 return new DepthwiseDepthfirst<uint8_t, int8_t>(strat, args, qp); in __anon3c4de31b0502()
[all …]
|
| D | depthwise_depthfirst.hpp | 166 …DepthwiseDepthfirstCommon(StratType *const strat, const DepthwiseArgs &args, const OutputStage &os) in DepthwiseDepthfirstCommon() argument
167 : DepthfirstDriver<TInput, TWeight, TOutput>(strat, args), m_os(os) in DepthwiseDepthfirstCommon()
233 …static inline void indirect(const Strat *strat, const Workspace *ws, const OutputStage &, const vo… in indirect()
235 strat->get_indirect_kernel()( in indirect()
245 const Strat *strat, const Workspace *ws, const OutputStage &, in direct()
252 strat->get_direct_kernel()( in direct()
267 …static inline void indirect(const Strat *strat, const Workspace *ws, const arm_gemm::Requantize32 … in indirect()
269 strat->get_kernel()( in indirect()
332 DepthwiseDepthfirst(StratType *const strat, const DepthwiseArgs &args, const OutputStage &os = {}) in DepthwiseDepthfirst() argument
333 : Parent(strat, args, os), m_bias(nullptr)
[all …]
|
| D | depthwise_depthfirst_multiplier.hpp | 334 const DepthwiseArgs &args, const WorkspaceType *ws, const Type *strat, in execute()
339 strat->get_kernel()( in execute()
355 const DepthwiseArgs &args, const WorkspaceType *ws, const Type *strat, in execute()
360 strat->get_kernel()( in execute()
364 strat->get_kernel_rows() * strat->get_kernel_cols(), in execute()
378 const DepthwiseArgs &args, const WorkspaceType *ws, const Type *strat, in execute()
383 strat->get_kernel()( in execute()
399 const DepthwiseArgs &args, const WorkspaceType *ws, const Type *strat, in execute()
409 strat->get_kernel()( in execute()
413 strat->get_kernel_rows() * strat->get_kernel_cols(), in execute()
[all …]
|
| D | depthwise_depthfirst_generic.hpp | 80 KernelStrategyType *strat, unsigned int n_output_rows, unsigned int n_output_cols, in GenericDepthfirstStrategy() argument
88 m_strategy(strat) in GenericDepthfirstStrategy()
138 const StratType *strat, const WorkspaceType *ws, const Nothing &, in execute()
143 strat->get_kernel()( in execute()
158 const StratType *strat, const WorkspaceType *ws, const arm_gemm::Requantize32 &qp, in execute()
163 strat->get_kernel()( in execute()
217 …DepthwiseDepthfirstGeneric(StratType *const strat, const DepthwiseArgs &args, const OutputStage &o… in DepthwiseDepthfirstGeneric() argument
218 : Parent(strat, args, os)
|
| /external/ComputeLibrary/src/core/NEON/kernels/arm_conv/pooling/ |
| D | pooling_fp32.cpp | 63 auto strat = new cpp_nhwc_1x1_stride_any_depthfirst<float>(args.cpu_info); in __anon663237960202() local
64 return new PoolingDepthfirstGeneric<float, float, Nothing>(strat, args); in __anon663237960202()
78 auto strat = new sme_fp32_nhwc_max_2x2_s1_output2x2_depthfirst(args.cpu_info); in __anon663237960402() local
79 return new PoolingDepthfirst<float>(strat, args); in __anon663237960402()
91 auto strat = new sme_fp32_nhwc_avg_3x3_s1_output2x2_depthfirst(args.cpu_info); in __anon663237960602() local
92 return new PoolingDepthfirst<float>(strat, args); in __anon663237960602()
103 auto strat = new sme_fp32_nhwc_avg_generic_depthfirst(args.cpu_info); in __anon663237960802() local
104 return new PoolingDepthfirstGeneric<float>(strat, args); in __anon663237960802()
115 auto strat = new sme_fp32_nhwc_max_generic_depthfirst(args.cpu_info); in __anon663237960a02() local
116 return new PoolingDepthfirstGeneric<float>(strat, args); in __anon663237960a02()
[all …]
|
| D | pooling_fp16.cpp | 66 auto strat = new cpp_nhwc_1x1_stride_any_depthfirst<__fp16>(args.cpu_info); in __anon61d048580202() local
67 return new PoolingDepthfirstGeneric<__fp16>(strat, args); in __anon61d048580202()
81 auto strat = new sme_fp16_nhwc_max_2x2_s1_output2x2_depthfirst(args.cpu_info); in __anon61d048580402() local
82 return new PoolingDepthfirst<__fp16>(strat, args); in __anon61d048580402()
94 auto strat = new sme_fp16_nhwc_avg_3x3_s1_output2x2_depthfirst(args.cpu_info); in __anon61d048580602() local
95 return new PoolingDepthfirst<__fp16>(strat, args); in __anon61d048580602()
106 auto strat = new sme_fp16_nhwc_avg_generic_depthfirst(args.cpu_info); in __anon61d048580802() local
107 return new PoolingDepthfirstGeneric<__fp16>(strat, args); in __anon61d048580802()
118 auto strat = new sme_fp16_nhwc_max_generic_depthfirst(args.cpu_info); in __anon61d048580a02() local
119 return new PoolingDepthfirstGeneric<__fp16>(strat, args); in __anon61d048580a02()
[all …]
|
| D | pooling_s8.cpp | 62 auto strat = new cpp_nhwc_1x1_stride_any_depthfirst<int8_t>(args.cpu_info); in __anonad6052460202() local
63 return new PoolingDepthfirstGeneric<int8_t>(strat, args); in __anonad6052460202()
77 auto strat = new sme_s8_nhwc_max_2x2_s1_output2x2_depthfirst(args.cpu_info); in __anonad6052460402() local
78 return new PoolingDepthfirst<int8_t>(strat, args); in __anonad6052460402()
89 auto strat = new sme_s8_nhwc_avg_generic_depthfirst(args.cpu_info); in __anonad6052460602() local
90 return new PoolingDepthfirstGeneric<int8_t>(strat, args); in __anonad6052460602()
101 auto strat = new sme_s8_nhwc_max_generic_depthfirst(args.cpu_info); in __anonad6052460802() local
102 return new PoolingDepthfirstGeneric<int8_t>(strat, args); in __anonad6052460802()
116 auto strat = new sve_s8_nhwc_max_2x2_s1_output2x2_depthfirst(args.cpu_info); in __anonad6052460a02() local
117 return new PoolingDepthfirst<int8_t>(strat, args); in __anonad6052460a02()
[all …]
|
| D | pooling_u8.cpp | 62 auto strat = new cpp_nhwc_1x1_stride_any_depthfirst<uint8_t>(args.cpu_info); in __anonb20aa3880202() local
63 return new PoolingDepthfirstGeneric<uint8_t>(strat, args); in __anonb20aa3880202()
77 auto strat = new sme_u8_nhwc_max_2x2_s1_output2x2_depthfirst(args.cpu_info); in __anonb20aa3880402() local
78 return new PoolingDepthfirst<uint8_t>(strat, args); in __anonb20aa3880402()
96 auto strat = new sme_u8_nhwc_avg_generic_depthfirst(args.cpu_info); in __anonb20aa3880602() local
97 return new PoolingDepthfirstGeneric<uint8_t>(strat, args); in __anonb20aa3880602()
108 auto strat = new sme_u8_nhwc_max_generic_depthfirst(args.cpu_info); in __anonb20aa3880802() local
109 return new PoolingDepthfirstGeneric<uint8_t>(strat, args); in __anonb20aa3880802()
123 auto strat = new sve_u8_nhwc_max_2x2_s1_output2x2_depthfirst(args.cpu_info); in __anonb20aa3880a02() local
124 return new PoolingDepthfirst<uint8_t>(strat, args); in __anonb20aa3880a02()
[all …]
|
| D | pooling_u8q.cpp | 59 auto strat = new sme_u8q_nhwc_avg_generic_depthfirst(args.cpu_info); in __anonf7fe37590202() local
60 return new PoolingDepthfirstGeneric<uint8_t, uint8_t, Requantize32>(strat, args, rq); in __anonf7fe37590202()
71 auto strat = new sme_u8q_nhwc_max_generic_depthfirst(args.cpu_info); in __anonf7fe37590402() local
72 return new PoolingDepthfirstGeneric<uint8_t, uint8_t, Requantize32>(strat, args, rq); in __anonf7fe37590402()
85 auto strat = new sve_u8q_nhwc_avg_generic_depthfirst(args.cpu_info); in __anonf7fe37590602() local
86 return new PoolingDepthfirstGeneric<uint8_t, uint8_t, Requantize32>(strat, args, rq); in __anonf7fe37590602()
97 auto strat = new sve_u8q_nhwc_max_generic_depthfirst(args.cpu_info); in __anonf7fe37590802() local
98 return new PoolingDepthfirstGeneric<uint8_t, uint8_t, Requantize32>(strat, args, rq); in __anonf7fe37590802()
110 auto strat = new a64_u8q_nhwc_avg_generic_depthfirst(args.cpu_info); in __anonf7fe37590a02() local
111 return new PoolingDepthfirstGeneric<uint8_t, uint8_t, Requantize32>(strat, args, rq); in __anonf7fe37590a02()
[all …]
|
| D | pooling_s8q.cpp | 59 auto strat = new sme_s8q_nhwc_avg_generic_depthfirst(args.cpu_info); in __anon5e09bdd70202() local
60 return new PoolingDepthfirstGeneric<int8_t, int8_t, Requantize32>(strat, args, rq); in __anon5e09bdd70202()
71 auto strat = new sme_s8q_nhwc_max_generic_depthfirst(args.cpu_info); in __anon5e09bdd70402() local
72 return new PoolingDepthfirstGeneric<int8_t, int8_t, Requantize32>(strat, args, rq); in __anon5e09bdd70402()
85 auto strat = new sve_s8q_nhwc_avg_generic_depthfirst(args.cpu_info); in __anon5e09bdd70602() local
86 return new PoolingDepthfirstGeneric<int8_t, int8_t, Requantize32>(strat, args, rq); in __anon5e09bdd70602()
97 auto strat = new sve_s8q_nhwc_max_generic_depthfirst(args.cpu_info); in __anon5e09bdd70802() local
98 return new PoolingDepthfirstGeneric<int8_t, int8_t, Requantize32>(strat, args, rq); in __anon5e09bdd70802()
110 auto strat = new a64_s8q_nhwc_avg_generic_depthfirst(args.cpu_info); in __anon5e09bdd70a02() local
111 return new PoolingDepthfirstGeneric<int8_t, int8_t, Requantize32>(strat, args, rq); in __anon5e09bdd70a02()
[all …]
|
| D | pooling_depthfirst_generic.hpp | 121 GenericDepthfirstWrapper(const StratType *strat, const PoolingArgs &args) in GenericDepthfirstWrapper() argument
122 : m_strat(strat), window_rows(args.pool_window.rows), window_cols(args.pool_window.cols) in GenericDepthfirstWrapper()
276 const IGenericDepthfirstStrategy<TInput, TOutput, OutputStage> *strat, in PoolingDepthfirstGeneric() argument
281 new GenericDepthfirstWrapper<TInput, TOutput, OutputStage>(strat, args),
|
| /external/tensorflow/tensorflow/python/distribute/ |
| D | random_generator_test.py | 44 def get_num_local_replicas(strat, values=None): argument
45 strat_name = type(strat).__name__
48 values = strat.run(lambda: constant_op.constant(0))
49 values = strat.experimental_local_results(values)
52 return strat.num_replicas_in_sync
66 def run_on_strategy(replica_fn, strat, coord): argument
68 return strat.run(replica_fn)
105 strat = MirroredStrategy(devices=["cpu:0", "cpu:1"])
106 with strat.scope():
114 results = strat.extended.call_for_each_replica(fn=f)
[all …]
|
| /external/ComputeLibrary/src/core/NEON/kernels/arm_gemm/ |
| D | gemv_pretransposed.hpp | 46 const strategy &strat,
57 const strategy &strat, in run() argument
64 strat.kernel(A_ptr, B_ptr, C_ptr, N, K, bias, act, Accumulate); in run()
70 const strategy &strat, in run() argument
77 strat.kernel(A_ptr, B_ptr, C_ptr, N, K, &qp, col_bias + col_base, col_base); in run()
147 strategy strat(_args._ci); in execute() local
178 … run_gemv_kernel<OutputStage>::run(strat, this->_Aptr + (multi * this->_A_multi_stride) + k0, in execute()
225 strategy strat(_args._ci); in pretranspose_B_array() local
228 …strat.transforms.PrepareB(B_buffer + (multi * _buffer_per_multi), B + (multi * B_multi_stride), ld… in pretranspose_B_array()
|
| D | gemm_hybrid_indirect.hpp | 66 …const strategy &strat, unsigned int num_strings, const unsigned int *string_ptr, IndirectInputArg<…
77 …const strategy &strat, unsigned int num_strings, const unsigned int *string_ptr, IndirectInputArg<… in run() argument
97 …strat.kernel(num_strings, string_ptr, A_arg, M, N_bulk, b_ptr, output_arg, bias_ptr, act, accumula… in run()
111 …strat.kernel(num_strings, string_ptr, A_arg, M, N_remainder, b_ptr + (N_bulk * kern_k), offset_out… in run()
113 … strat.kernel(num_strings, string_ptr, A_arg, M, N, b_ptr, output_arg, bias_ptr, act, accumulate); in run()
123 …const strategy &strat, unsigned int num_strings, const unsigned int *string_ptr, IndirectInputArg<… in run() argument
143 …strat.kernel(num_strings, string_ptr, A_arg, M, N_bulk, b_ptr, b_stride, output_arg, bias_ptr, act… in run()
157 strat.kernel(num_strings, string_ptr, A_arg, M, N_remainder, in run()
161 …strat.kernel(num_strings, string_ptr, A_arg, M, N, b_ptr, b_stride, output_arg, bias_ptr, act, acc… in run()
171 …const strategy &strat, unsigned int num_strings, const unsigned int *string_ptr, IndirectInputArg<… in run() argument
[all …]
|
| D | gemm_interleaved.hpp | 70 strategy &strat, const To *a_ptr, const To *b_panel, size_t b_stride, Tri *c_panel,
84 strategy &strat, const To *a_ptr, const To *b_panel, size_t, Tri *c_panel, in run() argument
96 strat.kernel(a_ptr, b_panel, c_panel, 1, bblocks, kern_k); in run()
103 … strat.transforms.Merge(c_ptr, c_panel, ldc, m_0, m_max, n_0, n_max, biasptr, act, accumulate); in run()
114 strategy &strat, const To *a_ptr, const To *b_panel, size_t b_stride, Tri *c_panel, in run() argument
125 strat.kernel(a_ptr, b_panel, b_stride, c_panel, 1, (n_max - n_0), kern_k); in run()
133 … strat.transforms.Merge(c_ptr, c_panel, ldc, m_0, m_max, n_0, n_max, biasptr, act, accumulate); in run()
144 strategy &strat, const To *a_ptr, const To *b_panel, size_t, Tri *, in run() argument
164 strat.kernel(// A and B pointers are just the packed panels. in run()
183 strategy &strat, const To *a_ptr, const To *b_panel, size_t, Tri *, in run() argument
[all …]
|
| D | gemm_interleaved_pretransposed_2d.hpp | 184 strategy strat(_ci); in execute_pretranspose() local
225 kern_k *= strat.k_unroll(); in execute_pretranspose()
230 … b_panel = b_panel_start + (b_thread_start_offset * strat.out_width() * kern_k); in execute_pretranspose()
246 strat.transforms.PrepareA( in execute_pretranspose()
271 strat.kernel(a_ptr, b_panel, c_panel, 1, bblocks, kern_k); in execute_pretranspose()
288 strat.transforms.Merge( in execute_pretranspose()
302 b_panel += (bblocks * strat.out_width() * kern_k); in execute_pretranspose()
485 strategy strat(_ci); in pretranspose_B_array() local
499 strat.transforms.PrepareB(buffer, B + (current.multi() * B_multi_stride), ldb, in pretranspose_B_array()
|
| D | gemm_hybrid_quantized_inline.hpp | 163 strategy strat(_ci); in execute() local
199 …strat.kernel(this->_Aptr + (multi * this->_A_multi_stride) + (batch * this->_A_batch_stride) + (m_… in execute()
236 strategy strat(_ci); in pretranspose_B_array() local
248 strat.transforms.PrepareB( buffer, B + (multi * B_multi_stride), ldb, in pretranspose_B_array()
|
| D | gemm_hybrid.hpp | 153 strategy strat(_ci); in execute() local
193 …strat.kernel(this->_Aptr + (multi * this->_A_multi_stride) + (batch * this->_A_batch_stride) + (m_… in execute()
227 strategy strat(_ci); in pretranspose_B_array() local
239 strat.transforms.PrepareB( buffer, B + (multi * B_multi_stride), ldb, in pretranspose_B_array()
|
| D | gemm_hybrid_quantized.hpp | 179 strategy strat(_ci); in execute() local
221 …strat.kernel(this->_Aptr + (multi * this->_A_multi_stride) + (batch * this->_A_batch_stride) + (m_… in execute()
286 strategy strat(_ci); in pretranspose_B_array() local
298 strat.transforms.PrepareB( buffer, B + (multi * B_multi_stride), ldb, in pretranspose_B_array()
|
| /external/google-breakpad/src/processor/ |
| D | range_map.h | 72 void SetMergeStrategy(MergeRangeStrategy strat) { merge_strategy_ = strat; } in SetMergeStrategy() argument
|