Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/cc/framework/
Dgradient_checker.cc145 for (int y_idx = 0; y_idx < y_num; y_idx++) { in ComputeTheoreticalJacobianTranspose() local
146 auto dy_data_flat = dy_datas[y_idx].flat<Y_T>(); in ComputeTheoreticalJacobianTranspose()
147 const int64 dy_size = y_shapes[y_idx].num_elements(); in ComputeTheoreticalJacobianTranspose()
162 auto jacobian = (*jacobian_ts)[x_idx * y_num + y_idx].matrix<JAC_T>(); in ComputeTheoreticalJacobianTranspose()
189 for (int y_idx = 0; y_idx < y_datas->size(); y_idx++) { in EvaluateGraph() local
191 Tensor y_data = (*y_datas)[y_idx]; in EvaluateGraph()
197 (*y_datas)[y_idx] = tensor::DeepCopy(y_data); in EvaluateGraph()
244 for (int y_idx = 0; y_idx < y_num; y_idx++) { in ComputeNumericJacobianTranspose() local
247 auto y_pos_flat = y_pos[y_idx].flat<Y_T>(); in ComputeNumericJacobianTranspose()
248 auto y_neg_flat = y_neg[y_idx].flat<Y_T>(); in ComputeNumericJacobianTranspose()
[all …]
/external/libaom/libaom/av1/encoder/
Dencodeframe.c2036 int y_idx = (i >> 1) * hbs; in rd_use_partition() local
2039 if ((mi_row + y_idx >= mi_params->mi_rows) || in rd_use_partition()
2046 mi_row + y_idx, mi_col + x_idx, subsize, &tmp_rdc.rate, in rd_use_partition()
2089 int y_idx = (i >> 1) * hbs; in rd_use_partition() local
2092 if ((mi_row + y_idx >= mi_params->mi_rows) || 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()
2112 encode_sb(cpi, td, tile_data, tp, mi_row + y_idx, mi_col + x_idx, in rd_use_partition()
2171 int y_idx = (i >> 1) * hbs; in is_leaf_split_partition() local
2172 if ((mi_row + y_idx >= cm->mi_params.mi_rows) || in is_leaf_split_partition()
2175 if (get_partition(cm, mi_row + y_idx, mi_col + x_idx, subsize) != in is_leaf_split_partition()
[all …]
Dpartition_strategy.c974 const int y_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.c2811 int y_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()
2855 int y_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()
2880 encode_sb(cpi, td, tile_info, tp, mi_row + y_idx, mi_col + x_idx, 0, in rd_use_partition()
2883 pl = partition_plane_context(xd, mi_row + y_idx, mi_col + x_idx, in rd_use_partition()
3631 const int y_idx = (i >> 1) * bs / 2; in ml_predict_var_rd_paritioning() local
3632 const int src_offset = y_idx * src_stride + x_idx; in ml_predict_var_rd_paritioning()
[all …]
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/
Dtf_ops_n_z.cc2435 const int64_t y_idx = e.index(); in Verify() local
2436 const int64_t y_dim = y_type.getDimSize(y_idx); in Verify()
2442 << "requires y.shape[" << y_idx << "] (" << y_dim << ") " in Verify()