Home
last modified time | relevance | path

Searched refs:x_idx (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/cc/framework/
Dgradient_checker.cc160 for (int x_idx = 0; x_idx < x_num; x_idx++) { in ComputeTheoreticalJacobianTranspose() local
161 const int64 x_size = x_shapes[x_idx].num_elements(); in ComputeTheoreticalJacobianTranspose()
162 auto jacobian = (*jacobian_ts)[x_idx * y_num + y_idx].matrix<JAC_T>(); in ComputeTheoreticalJacobianTranspose()
163 auto dx_flat = dxout[x_idx].flat<X_T>(); in ComputeTheoreticalJacobianTranspose()
190 for (int x_idx = 0; x_idx < x_datas->size(); x_idx++) { in EvaluateGraph() local
192 if (y_data.SharesBufferWith((*x_datas)[x_idx])) { in EvaluateGraph()
221 for (int x_idx = 0; x_idx < x_num; x_idx++) { in ComputeNumericJacobianTranspose() local
222 auto x_data_flat = (*x_datas)[x_idx].flat<X_T>(); in ComputeNumericJacobianTranspose()
223 const int64 x_size = x_shapes[x_idx].num_elements(); in ComputeNumericJacobianTranspose()
251 auto jacobian = (*jacobian_ts)[x_idx * y_num + y_idx].matrix<JAC_T>(); in ComputeNumericJacobianTranspose()
[all …]
/external/libaom/libaom/av1/encoder/
Dencodeframe.c532 int i, x_idx, y; in update_state() local
608 for (x_idx = 0; x_idx < mi_width; x_idx++) { in update_state()
609 if ((xd->mb_to_right_edge >> (3 + MI_SIZE_LOG2)) + mi_width > x_idx && in update_state()
611 xd->mi[x_idx + y * mis] = mi_addr; in update_state()
2035 int x_idx = (i & 1) * hbs; in rd_use_partition() local
2040 (mi_col + x_idx >= mi_params->mi_cols)) in rd_use_partition()
2046 mi_row + y_idx, mi_col + x_idx, subsize, &tmp_rdc.rate, in rd_use_partition()
2088 int x_idx = (i & 1) * hbs; in rd_use_partition() local
2093 (mi_col + x_idx >= mi_params->mi_cols)) in rd_use_partition()
2098 pick_sb_modes(cpi, tile_data, x, mi_row + y_idx, mi_col + x_idx, &tmp_rdc, in rd_use_partition()
[all …]
Dpartition_strategy.c973 const int x_idx = (i & 1) * bw / 2; in av1_ml_prune_rect_partition() local
975 buf.buf = x->plane[0].src.buf + x_idx + y_idx * buf.stride; in av1_ml_prune_rect_partition()
/external/libvpx/libvpx/vp9/encoder/
Dvp9_encodeframe.c1796 int i, x_idx, y; in update_state() local
1858 for (x_idx = 0; x_idx < mi_width; x_idx++) in update_state()
1859 if ((xd->mb_to_right_edge >> (3 + MI_SIZE_LOG2)) + mi_width > x_idx && in update_state()
1861 xd->mi[x_idx + y * mis] = mi_addr; in update_state()
2810 int x_idx = (i & 1) * (mi_step >> 1); in rd_use_partition() local
2814 if ((mi_row + y_idx >= cm->mi_rows) || (mi_col + x_idx >= cm->mi_cols)) in rd_use_partition()
2819 tp, mi_row + y_idx, mi_col + x_idx, subsize, in rd_use_partition()
2854 int x_idx = (i & 1) * (mi_step >> 1); in rd_use_partition() local
2860 if ((mi_row + y_idx >= cm->mi_rows) || (mi_col + x_idx >= cm->mi_cols)) in rd_use_partition()
2865 rd_pick_sb_modes(cpi, tile_data, x, mi_row + y_idx, mi_col + x_idx, in rd_use_partition()
[all …]
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/
Dtf_ops_n_z.cc2437 const int64_t x_idx = e.value().getSExtValue(); in Verify() local
2438 const int64_t x_dim = x_type.getDimSize(x_idx); in Verify()
2443 << "to be equal to x.shape[perm[" << x_idx << "]] " in Verify()