Searched refs:split_axis (Results 1 – 5 of 5) sorted by relevance
413 split_axis = 1414 size_splits = [1] * input_shape[split_axis]416 y = array_ops.split(x, num_or_size_splits=size_splits, axis=split_axis)418 for i in range(input_shape[split_axis]):430 split_axis = 1433 y = array_ops.split(x, num_or_size_splits=size_splits, axis=split_axis)
271 const unsigned int split_axis = 0; in add_lstm_cell() local274 …der::add_split_node(graph.graph(), split_params, { graph.tail_node(), 0 }, num_splits, split_axis); in add_lstm_cell()
976 const auto split_axis = in removeSequenceSplitConcat() local995 if (split_axis != concat_axis) { in removeSequenceSplitConcat()
1555 ushort split_axis = locals->axis_and_left_count[split_index].x;1598 float c = my_box->lower[split_axis] + my_box->upper[split_axis];1599 float Csplit = split_box->lower[split_axis] + split_box->upper[split_axis];
3017 ushort split_axis = locals->axis_and_left_count[split_index].x;3055 float c = my_box->lower[split_axis] + my_box->upper[split_axis];3056 float Csplit = split_box->lower[split_axis] + split_box->upper[split_axis];