Searched refs:bw_scratch_shape_ (Results 1 – 2 of 2) sorted by relevance
427 bw_scratch_shape_.dimensions = {n_batch, n_bw_cell * 3}; in Prepare()430 bw_scratch_shape_.dimensions = {n_batch, n_bw_cell * 4}; in Prepare()432 fw_scratch_shape_.type = bw_scratch_shape_.type = inputShape.type; in Prepare()433 fw_scratch_shape_.offset = bw_scratch_shape_.offset = inputShape.offset; in Prepare()434 fw_scratch_shape_.scale = bw_scratch_shape_.scale = inputShape.scale; in Prepare()534 std::vector<float> bw_scratch_buffer(getNumberOfElements(bw_scratch_shape_)); in Eval()659 std::vector<_Float16> bw_scratch_buffer(getNumberOfElements(bw_scratch_shape_)); in Eval()
158 Shape bw_scratch_shape_; variable